Home
last modified time | relevance | path

Searched defs:input_devices (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute_node_test.cc34 TestKernelAndDeviceFunc(std::vector<Device*> input_devices, in TestKernelAndDeviceFunc()
111 std::vector<Device*> input_devices; in TEST() local
Dkernel_and_device.h272 std::vector<Device*> input_devices, in KernelAndDeviceFunc()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/saved_model/
Dsaved_model.h54 llvm::ArrayRef<llvm::StringRef> input_devices; member
Dsaved_model.cc85 llvm::SmallVector<llvm::StringRef, 4> input_devices; in MapFunctionSignaturesFromTFSavedModelMLIR() local
/external/tensorflow/tensorflow/core/tfrt/eager/
Dfunction_cache.cc51 tfrt::ArrayRef<const Device*> input_devices, in GetOrAddFunction()
Dtransform_graph_function.cc41 tfrt::ArrayRef<const tfrt::Device*> input_devices, in TransformGraphFunction()
Dfunction_cache_test.cc228 const Device* input_devices[2] = {&device, &device}; in TEST_P() local
Dc_api_tfrt.cc1528 llvm::SmallVector<const tfrt::Device*, 4> input_devices; in Initialize() local
/external/tensorflow/tensorflow/core/tfrt/saved_model/
Dsaved_model.h73 std::vector<std::string> input_devices; member
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dcluster_tf_ops_pass.cc98 llvm::SmallVector<std::string, 4> input_devices; member
/external/tensorflow/tensorflow/compiler/xla/python/
Dpmap_lib.cc387 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/
Dinline_function_utils.cc124 std::vector<string> input_devices(caller.in_edges().size()); in InputDevices() local
Dprocess_function_library_runtime_test.cc687 const string& target, const std::vector<string>& input_devices, in MakeOptions()
Dprocess_function_library_runtime.cc459 const std::vector<string>& input_devices, in PinArgsAndRets()
/external/tensorflow/tensorflow/core/framework/
Dfunction.h659 std::vector<string> input_devices; member