Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dtpu_rewrite_device_util.cc547 Device parsed_device; in IsTPUDevice() local
555 Device parsed_device; in IsTPUReplicatedCore() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dsharding_util.cc64 DeviceNameUtils::ParsedName parsed_device; in ParseShardingFromDevice() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/
H A Dschema_type_parser.cpp253 auto parsed_device = tryToParseDeviceType(); in parseRefinedTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_dynamic_layout_pass.cc110 tensorflow::DeviceNameUtils::ParsedName parsed_device; in IsSupportedInputOp() local
H A Dtpu_space_to_depth_pass.cc369 tensorflow::DeviceNameUtils::ParsedName parsed_device; in IsSupportedHostInputOp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dcollective_param_resolver_local.cc87 DeviceNameUtils::ParsedName parsed_device; in TaskNameFromDeviceName() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_functional_ops.cc2566 DeviceNameUtils::ParsedName parsed_device; in InstantiateFunctionsFromSubgraphs() local