Searched refs:reduce_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_padder_test.cc | 74 auto reduce_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
88 reduce_shape, negate, init, {0, 2}, GetScalarAddComputation())); in TEST_F()
191 auto reduce_shape = ShapeUtil::MakeShape(F32, {3, 5}); in TEST_F() local
201 reduce_shape, input, init, window, GetScalarAddComputation())); in TEST_F()
|
| D | dynamic_dimension_inference_test.cc | 181 auto reduce_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
195 reduce_shape, negate, init, {0, 2}, GetAdd())); in TEST_F()
213 auto reduce_shape = ShapeUtil::MakeShape(F32, {1, 2}); in TEST_F() local
227 HloInstruction::CreateReduce(reduce_shape, negate, init, {1}, GetAdd())); in TEST_F()
|
| D | algebraic_simplifier.cc | 3134 Shape reduce_shape = ShapeUtil::FilterDimensions( in HandleReduceWindow() local
3141 /*shape=*/reduce_shape, in HandleReduceWindow()
|
| D | algebraic_simplifier_test.cc | 1737 Shape reduce_shape = ShapeUtil::MakeShape(F32, {kParamLength}); in TEST_F() local
1742 reduce_shape, Concatenate, zero, {1, 2}, add_computation)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | gpu_ldg_test.cc | 115 auto reduce_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
119 reduce_shape, in TEST_F()
126 HloInstruction::CreateUnary(reduce_shape, HloOpcode::kSin, reduce)); in TEST_F()
|