Searched defs:reduce_shape (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | dynamic_dimension_inference_test.cc | 211 auto reduce_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
243 auto reduce_shape = ShapeUtil::MakeShape(F32, {1, 2}); in TEST_F() local
276 auto reduce_shape = ShapeUtil::MakeShape(F32, {1, 2}); in TEST_F() local
|
| H A D | tree_reduction_rewriter.cc | 49 const Shape &reduce_shape = hlo->shape(); in HandleReduce() local
|
| H A D | reduce_decomposer.cc | 136 Shape reduce_shape = reduce->shape(); in ExpectedOutputShape() local
|
| H A D | dynamic_padder_test.cc | 158 auto reduce_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
419 auto reduce_shape = ShapeUtil::MakeShape(F32, {3, 5}); in TEST_F() local
|
| H A D | algebraic_simplifier.cc | 5762 Shape reduce_shape = ShapeUtil::DeleteDimensions(effective_reduce_dims, in HandleReduceWindow() local
|
| H A D | algebraic_simplifier_test.cc | 2679 Shape reduce_shape = ShapeUtil::MakeShape(F32, {kParamLength}); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | reduction_degenerate_dim_remover.cc | 49 const Shape &reduce_shape = instr->shape().IsTuple() in HandleReduce() local
|
| H A D | reduction_layout_normalizer.cc | 72 const Shape &reduce_shape = in HandleReduce() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| H A D | gpu_ldg_test.cc | 129 auto reduce_shape = ShapeUtil::MakeShape(F32, {32}); in TEST_F() local
|