Searched defs:input_device (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute_node.cc | 22 bool ExecuteNodeArgs::IsRemote(EagerContext* ctx, Device* input_device, in IsRemote()
37 Device* input_device, in InitPackedHandle()
|
| D | execute.cc | 341 Device* input_device; in GetDeviceForInput() local
978 Device* input_device; in ExtractFunctionInputInfo() local
1565 tensorflow::Device* input_device = input->device(); in EagerRemoteExecute() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/saved_model/ |
| D | saved_model.cc | 98 if (auto input_device = in MapFunctionSignaturesFromTFSavedModelMLIR() local
|
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | dtensor_device.cc | 589 const char* input_device = TFE_TensorHandleDeviceName(input, status); in FetchLayout() local
606 const char* input_device = TFE_TensorHandleDeviceName(input, status); in Unpack() local
707 const char* input_device = TFE_TensorHandleDeviceName(input_tensor, status); in CopyToMesh() local
864 const char* input_device = TFE_TensorHandleDeviceName(input, status); in Pack() local
1013 const char* input_device = in SparsePack() local
1064 const char* input_device = TFE_TensorHandleDeviceName(input, status); in IsSparseDTensor() local
1876 const char* input_device = TFE_TensorHandleDeviceName(input, status); in Execute() local
|
| D | dtensor_device_util.cc | 276 const char* input_device = TFE_TensorHandleDeviceName(tensor, status); in Broadcast() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | dependency_optimizer.cc | 672 string input_device = input->device(); in GroupCrossDeviceControlEdges() local
719 string input_device = input->device(); in GroupCrossDeviceControlEdges() local
|
| D | remapper_test.cc | 1375 const string input_device = in RunTest() local
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | custom_device_testutil.cc | 127 const char* input_device = TFE_TensorHandleDeviceName(input, s); in LoggingDeviceExecute() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | inline_function_utils.cc | 129 const string& input_device = edge->src()->has_assigned_device_name() in InputDevices() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler.cc | 1037 auto& input_device = device_[input_state.device_name]; in MarkNodeExecuted() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | rnn_cell_test.py | 2467 input_device=None): argument
|