Home
last modified time | relevance | path

Searched defs:while_output (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Ddtensor_allreduce_sum_optimization.cc263 const int index, mlir::TF::WhileRegionOp while_op, mlir::Value while_output, in CheckWhileLoopOptimizationCriteria()
379 mlir::Value while_output = while_op.getResult(output_index); in ExtractAllReduceFromWhileOp() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc5951 SmallVector<Value, 2> while_output; in matchAndRewrite() local
6880 SmallVector<Value, 4> while_output; in QRBlock() local
6986 SmallVector<Value, 4> while_output; in ComputeWYRepresentation() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner_test.cc5560 const auto while_output = in TEST_F() local