MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / device

Method device

tensorflow/core/common_runtime/function.cc:280–280  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

278Env* FunctionLibraryRuntimeOverlay::env() { return base_flr_->env(); }
279
280Device* FunctionLibraryRuntimeOverlay::device() { return base_flr_->device(); }
281
282const Device* FunctionLibraryRuntimeOverlay::device() const {
283 return base_flr_->device();

Callers 15

_replicated_variable_getterFunction · 0.45
opt_variableFunction · 0.45
_clone_optimizerFunction · 0.45
build_infeed_from_input_specsMethod · 0.45
build_infeed_from_input_specsMethod · 0.45
_specialize_modelMethod · 0.45
fitMethod · 0.45
test_mpi_allreduce_gpuMethod · 0.45
do_periodic_resample_opFunction · 0.45
variableFunction · 0.45
testVariableWithDeviceFromScopeMethod · 0.45
testVariableWithDeviceFunctionMethod · 0.45

Calls

no outgoing calls

Tested by 15

test_mpi_allreduce_gpuMethod · 0.36
testVariableWithDeviceFromScopeMethod · 0.36
testVariableWithDeviceFunctionMethod · 0.36
testVariableWithReplicaDeviceSetterMethod · 0.36
testVariableWithVariableDeviceChooserMethod · 0.36
testVariableWithVariableDeviceChooserWithReplicaMethod · 0.36
testVariableGPUPlacementMethod · 0.36
testBasicsMethod · 0.36
testSameNameNoClobberingMethod · 0.36
testSameObjectOKMethod · 0.36
testSaveByDictMethod · 0.36
testRestoreOnCreateMethod · 0.36