Searched refs:reduce_result_shape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 2917 const Shape& reduce_result_shape = in HandleReduce() local
2923 ShapeUtil::IsZeroElementArray(reduce_result_shape)) { in HandleReduce()
2937 HloInstruction::CreateBroadcast(reduce_result_shape, init_value, {})); in HandleReduce()
2945 if (ShapeUtil::ElementsIn(reduce_result_shape) == in HandleReduce()
2959 reduce, HloInstruction::CreateReshape(reduce_result_shape, arg)); in HandleReduce()
2972 if (reduce_result_shape.rank() <= 1 && in HandleReduce()
2981 reduce_result_shape, arg->mutable_operand(0), init_value, in HandleReduce()
3011 reduce_result_shape, arg->mutable_operand(0), init_value, in HandleReduce()
3055 reduce_result_shape, arg->mutable_operand(0), init_value, in HandleReduce()
3070 HloInstruction::CreateReduce(reduce_result_shape, operand, init_value, in HandleReduce()
[all …]
|