| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | host_compute_ops.cc | 58 DeviceNameUtils::ParsedName parsed_name; in RecvAtHostOp() local
89 DeviceNameUtils::ParsedName parsed_name; in ComputeAsync() local
215 DeviceNameUtils::ParsedName parsed_name; in SendFromHostOp() local
244 DeviceNameUtils::ParsedName parsed_name; in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | virtual_placer.cc | 64 DeviceNameUtils::ParsedName parsed_name; in VirtualPlacer() local
95 DeviceNameUtils::ParsedName parsed_name; in VirtualPlacer() local
143 DeviceNameUtils::ParsedName parsed_name; in to_lfqn_or_empty() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | composite_device.cc | 28 DeviceNameUtils::ParsedName parsed_name = host_name; in MakeDevice() local
43 DeviceNameUtils::ParsedName parsed_name; in MakeDevice() local
|
| H A D | renamed_device.cc | 31 DeviceNameUtils::ParsedName parsed_name; in NewRenamedDevice() local
|
| H A D | collective_param_resolver_local.cc | 345 DeviceNameUtils::ParsedName parsed_name; in ParseRingOrder() local
378 DeviceNameUtils::ParsedName parsed_name; in OrderTaskDeviceMap() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | distributed_tpu_rewrite_helpers.cc | 60 const auto& parsed_name = device->parsed_name(); in GetSystemDevice() local
125 const auto& parsed_name = host_device->parsed_name(); in GetHostSystemDevices() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | implementation_selector.cc | 263 DeviceNameUtils::ParsedName parsed_name; in MaybeOptimizeFunctionCall() local
307 DeviceNameUtils::ParsedName parsed_name; in FindDeviceIndex() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| H A D | device_util_test.cc | 53 DeviceNameUtils::ParsedName parsed_name; in Make() local
198 DeviceNameUtils::ParsedName parsed_name; in TEST() local
|
| H A D | device_util.cc | 161 DeviceNameUtils::ParsedName parsed_name; in GetDeviceOrdinalFromDeviceString() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | device_name_utils.cc | 185 DeviceNameUtils::ParsedName* parsed_name) { in CompleteName()
227 ParsedName parsed_name; in CanonicalizeDeviceName() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/util/ |
| H A D | kernel_registry.cc | 44 tensorflow::DeviceNameUtils::ParsedName parsed_name; in TryFindKernelClass() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/backends/cpu/ |
| H A D | cpu_registration.cc | 37 auto& parsed_name = device->parsed_name(); in RegisterCpuOpHandler() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | rewrite_util.cc | 35 tensorflow::DeviceNameUtils::ParsedName parsed_name; in GetOpDevice() local
|
| H A D | fused_kernel_matcher.cc | 219 tensorflow::DeviceNameUtils::ParsedName parsed_name; in GetDevice() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
| H A D | context_distributed_manager.cc | 225 DeviceNameUtils::ParsedName parsed_name; in CreateRemoteContexts() local
322 DeviceNameUtils::ParsedName parsed_name; in UpdateRemoteContexts() local
|
| H A D | custom_device.cc | 82 DeviceNameUtils::ParsedName parsed_name; in ParsedName() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_graph.cc | 275 DeviceNameUtils::ParsedName parsed_name; in GetEngineInfo() local
667 DeviceNameUtils::ParsedName parsed_name; in GetDeviceAndAllocator() local
|
| H A D | utils.cc | 223 DeviceNameUtils::ParsedName parsed_name; in GetDeviceParsedName() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/backends/gpu/ |
| H A D | gpu_registration.cc | 37 auto& parsed_name = device->parsed_name(); in RegisterGpuOpHandler() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| H A D | optimize.cc | 35 DeviceNameUtils::ParsedName parsed_name; in matchAndRewrite() local
|
| H A D | cross_device_transfer.cc | 54 DeviceNameUtils::ParsedName parsed_name; in CanonicalizeDeviceName() local
|
| H A D | remote_run_encapsulate.cc | 161 DeviceNameUtils::ParsedName parsed_name; in runOnOperation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/ |
| H A D | list_flex_ops.cc | 62 tensorflow::DeviceNameUtils::ParsedName parsed_name; in FindTensorflowKernelClass() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | device.h | 66 const DeviceNameUtils::ParsedName& parsed_name() const { in parsed_name() function
|
| H A D | device_base.cc | 76 const DeviceNameUtils::ParsedName& DeviceBase::parsed_name() const { in parsed_name() function in tensorflow::DeviceBase
|