Searched defs:input_devices (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute_node_test.cc | 34 TestKernelAndDeviceFunc(std::vector<Device*> input_devices, in TestKernelAndDeviceFunc()
106 std::vector<Device*> input_devices; in TEST() local
|
| D | kernel_and_device.h | 267 std::vector<Device*> input_devices, in KernelAndDeviceFunc()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/saved_model/ |
| D | saved_model.h | 54 llvm::ArrayRef<llvm::StringRef> input_devices; member
|
| D | saved_model.cc | 85 llvm::SmallVector<llvm::StringRef, 4> input_devices; in MapFunctionSignaturesFromTFSavedModelMLIR() local
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | function_cache.cc | 51 tfrt::ArrayRef<const Device*> input_devices, in GetOrAddFunction()
|
| D | transform_graph_function.cc | 41 tfrt::ArrayRef<const tfrt::Device*> input_devices, in TransformGraphFunction()
|
| D | function_cache_test.cc | 228 const Device* input_devices[2] = {&device, &device}; in TEST_P() local
|
| D | c_api_tfrt.cc | 1466 tfrt::SmallVector<const tfrt::Device*, 4> input_devices; in Initialize() local
|
| /external/tensorflow/tensorflow/core/tfrt/saved_model/ |
| D | saved_model.h | 78 std::vector<std::string> input_devices; member
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | cluster_tf_ops_pass.cc | 97 llvm::SmallVector<std::string, 4> input_devices; member
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | inline_function_utils.cc | 124 std::vector<string> input_devices(caller.in_edges().size()); in InputDevices() local
|
| D | process_function_library_runtime_test.cc | 630 const string& target, const std::vector<string>& input_devices, in MakeOptions()
|
| D | process_function_library_runtime.cc | 422 const std::vector<string>& input_devices, in PinArgsAndRets()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 657 std::vector<string> input_devices; member
|