Home
last modified time | relevance | path

Searched refs:GetOutputTypesForLogicalDeviceComputation (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dxla_sharding_util.h57 mlir::LogicalResult GetOutputTypesForLogicalDeviceComputation(
Dxla_sharding_util.cc511 mlir::LogicalResult GetOutputTypesForLogicalDeviceComputation( in GetOutputTypesForLogicalDeviceComputation() function
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_rewrite_pass.cc445 auto result = tensorflow::GetOutputTypesForLogicalDeviceComputation( in BuildExecuteOp()
527 auto result = tensorflow::GetOutputTypesForLogicalDeviceComputation( in AddToParallelExecuteOp()