Searched defs:reduce_shape (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | reduction_degenerate_dim_remover.cc | 42 const Shape &reduce_shape = instr->shape(); in HandleReduce() local
|
| D | reduction_layout_normalizer.cc | 43 const Shape &reduce_shape = reduce->shape(); in HandleReduce() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tree_reduction_rewriter.cc | 48 const Shape &reduce_shape = hlo->shape(); in HandleReduce() local
|
| D | dynamic_dimension_inference_test.cc | 204 auto reduce_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
236 auto reduce_shape = ShapeUtil::MakeShape(F32, {1, 2}); in TEST_F() local
269 auto reduce_shape = ShapeUtil::MakeShape(F32, {1, 2}); in TEST_F() local
|
| D | dynamic_padder_test.cc | 116 auto reduce_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
347 auto reduce_shape = ShapeUtil::MakeShape(F32, {3, 5}); in TEST_F() local
|
| D | algebraic_simplifier.cc | 4818 Shape reduce_shape = ShapeUtil::FilterDimensions( in HandleReduceWindow() local
|
| D | algebraic_simplifier_test.cc | 2264 Shape reduce_shape = ShapeUtil::MakeShape(F32, {kParamLength}); in TEST_F() local
|
| /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
|