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.cc63 DeviceNameUtils::ParsedName parsed_device; in ParseShardingFromDevice() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_dynamic_layout_pass.cc127 tensorflow::DeviceNameUtils::ParsedName parsed_device; in IsSupportedInputOp() local
Dtpu_space_to_depth_pass.cc414 tensorflow::DeviceNameUtils::ParsedName parsed_device; in IsSupportedHostInputOp() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util.cc545 Device parsed_device; in IsTPUDevice() local
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local.cc79 DeviceNameUtils::ParsedName parsed_device; in TaskNameFromDeviceName() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc2805 DeviceNameUtils::ParsedName parsed_device; in CreateOrGetPerHostVariableCopy() local