Searched defs:input_devices (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute_node_test.cc | 34 TestKernelAndDeviceFunc(std::vector<Device*> input_devices, in TestKernelAndDeviceFunc()
111 std::vector<Device*> input_devices; in TEST() local
|
| D | kernel_and_device.h | 272 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 | 1528 llvm::SmallVector<const tfrt::Device*, 4> input_devices; in Initialize() local
|
| /external/tensorflow/tensorflow/core/tfrt/saved_model/ |
| D | saved_model.h | 73 std::vector<std::string> input_devices; member
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | cluster_tf_ops_pass.cc | 98 llvm::SmallVector<std::string, 4> input_devices; member
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | pmap_lib.cc | 387 auto input_devices = in PopulateCacheEntry() local
475 std::vector<xla::PjRtDevice*>& input_devices = cache_entry.devices; in Call() local
|
| /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 | 687 const string& target, const std::vector<string>& input_devices, in MakeOptions()
|
| D | process_function_library_runtime.cc | 459 const std::vector<string>& input_devices, in PinArgsAndRets()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 659 std::vector<string> input_devices; member
|