Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dxla_sharding_util.h42 mlir::LogicalResult ExtractInputsForLogicalDevices(
H A Dxla_sharding_util.cc204 mlir::LogicalResult ExtractInputsForLogicalDevices( in ExtractInputsForLogicalDevices() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_rewrite_pass.cc544 auto result = tensorflow::ExtractInputsForLogicalDevices( in AddToParallelExecuteOp()