Home
last modified time | relevance | path

Searched defs:parsed_device (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dsharding_util.cc64 DeviceNameUtils::ParsedName parsed_device; in ParseShardingFromDevice() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util.cc547 Device parsed_device; in IsTPUDevice() local
555 Device parsed_device; in IsTPUReplicatedCore() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_dynamic_layout_pass.cc110 tensorflow::DeviceNameUtils::ParsedName parsed_device; in IsSupportedInputOp() local
Dtpu_space_to_depth_pass.cc369 tensorflow::DeviceNameUtils::ParsedName parsed_device; in IsSupportedHostInputOp() local
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local.cc87 DeviceNameUtils::ParsedName parsed_device; in TaskNameFromDeviceName() local
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_functional_ops.cc2566 DeviceNameUtils::ParsedName parsed_device; in InstantiateFunctionsFromSubgraphs() local