Home
last modified time | relevance | path

Searched refs:ComponentFunctionData (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dprocess_function_library_runtime.h273 struct ComponentFunctionData { struct
332 std::unordered_map<string, ComponentFunctionData> glue_;
425 const ComponentFunctionData& comp_data,
430 const ComponentFunctionData& comp_data,
444 std::function<Status(const ComponentFunctionData& comp_data,
453 std::function<Status(const ComponentFunctionData& comp_data,
H A Dprocess_function_library_runtime.cc330 const ComponentFunctionData& component_data = pair.second; in GetHandleOnDevice()
1048 ComponentFunctionData* comp_data = &data->glue_[pair.first]; in InstantiateMultiDevice()
1063 ComponentFunctionData* comp_data = &data->glue_[pair.first]; in InstantiateMultiDevice()
1169 const ComponentFunctionData& comp_data = pair.second; in GetOutputDevices()
1263 std::function<Status(const ComponentFunctionData& comp_data, in RunMultiDeviceSync()
1292 const ComponentFunctionData& comp_data = data->glue_.at(target); in RunMultiDeviceSync()
1360 std::function<Status(const ComponentFunctionData& comp_data, in RunMultiDeviceAsync()
1387 const ComponentFunctionData& comp_data = pair.second; in RunMultiDeviceAsync()
1666 const ProcessFunctionLibraryRuntime::ComponentFunctionData& comp_data, in GetComponentArgs()
1698 const ProcessFunctionLibraryRuntime::ComponentFunctionData& comp_data, in GetComponentArgs()
[all …]