Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/dtensor/mlir/
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
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
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
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_test.cc5560 const auto while_output = in TEST_F() local