| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | reduction_layout_normalizer.cc | 43 const Shape &reduce_shape = reduce->shape(); in HandleReduce() local
45 if (!reduce_shape.IsArray()) { in HandleReduce()
57 std::vector<int64> new_reduce_shape_layout(reduce_shape.rank()); in HandleReduce()
59 LayoutUtil::MakeLogicalToPhysical(reduce_shape.layout()); in HandleReduce()
88 new_reduce_shape_layout[reduce_shape.rank() - physical_reduce_dim - 1] = in HandleReduce()
100 reduce_shape.element_type(), new_reduce_shape_data, in HandleReduce()
112 if (new_reduce_shape != reduce_shape) { in HandleReduce()
115 new_reduce = HloInstruction::CreateBitcast(reduce_shape, wrapped_reduce); in HandleReduce()
|
| D | reduction_degenerate_dim_remover.cc | 42 const Shape &reduce_shape = instr->shape(); in HandleReduce() local
52 ShapeUtil::DropDegenerateDimensions(reduce_shape); in HandleReduce()
70 HloInstruction::CreateBitcast(reduce_shape, reduced_op); in HandleReduce()
81 if (canonical_reduce_shape != reduce_shape) { in HandleReduce()
84 new_reduce = HloInstruction::CreateBitcast(reduce_shape, wrapped_reduce); in HandleReduce()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tree_reduction_rewriter.cc | 48 const Shape &reduce_shape = hlo->shape(); in HandleReduce() local
49 if (!reduce_shape.IsArray()) { in HandleReduce()
89 HloInstruction::CreateReduce(reduce_shape, reduce_window, initial_value, in HandleReduce()
|
| D | dynamic_dimension_inference_test.cc | 204 auto reduce_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
218 reduce_shape, negate, init, {0, 2}, GetAdd())); in TEST_F()
236 auto reduce_shape = ShapeUtil::MakeShape(F32, {1, 2}); in TEST_F() local
250 HloInstruction::CreateReduce(reduce_shape, negate, init, {1}, GetAdd())); in TEST_F()
269 auto reduce_shape = ShapeUtil::MakeShape(F32, {1, 2}); in TEST_F() local
293 ShapeUtil::MakeTupleShape({reduce_shape, reduce_shape}), in TEST_F()
|
| D | dynamic_padder_test.cc | 116 auto reduce_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
130 reduce_shape, negate, init, {0, 2}, GetScalarAddComputation())); in TEST_F()
347 auto reduce_shape = ShapeUtil::MakeShape(F32, {3, 5}); in TEST_F() local
357 reduce_shape, input, init, window, GetScalarAddComputation())); in TEST_F()
|
| D | algebraic_simplifier.cc | 4818 Shape reduce_shape = ShapeUtil::FilterDimensions( in HandleReduceWindow() local
4823 simplifier_->UpdateLayout(&reduce_shape); in HandleReduceWindow()
4826 /*shape=*/reduce_shape, in HandleReduceWindow()
|
| D | algebraic_simplifier_test.cc | 2264 Shape reduce_shape = ShapeUtil::MakeShape(F32, {kParamLength}); in TEST_F() local
2269 reduce_shape, Concatenate, zero, {1, 2}, add_computation)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | gpu_ldg_test.cc | 130 auto reduce_shape = ShapeUtil::MakeShape(F32, {32}); in TEST_F() local
134 reduce_shape, in TEST_F()
141 HloInstruction::CreateUnary(reduce_shape, HloOpcode::kSin, reduce)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 2742 auto reduce_shape, in HandleReduce()
2751 reduce_shape, input_hlos, inits, hlo->dimensions(), hlo->to_apply())); in HandleReduce()
2777 ShapeUtil::GetTupleElementShape(reduce_shape, i), local_reduce, in HandleReduce()
2796 reduce_shape, all_gathered_partial_results, inits, in HandleReduce()
|
| D | dot_handler.cc | 3296 TF_ASSIGN_OR_RETURN(auto reduce_shape, in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
3301 *reduce_shape.mutable_layout() = reduce_outside->shape().layout(); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
3345 reduce_shape, {operand0, operand1})); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
|