Searched refs:reduce_window_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 352 auto reduce_window_shape = new_convolution->shape(); in HandleBatchGroupCount() local
353 reduce_window_shape.set_dimensions(output_batch_dimension, 1); in HandleBatchGroupCount()
358 reduce_window_shape.set_element_type(F32); in HandleBatchGroupCount()
398 reduce_window_shape, new_filter, zero_scalar, window, in HandleBatchGroupCount()
|
| D | algebraic_simplifier_test.cc | 4383 const Shape reduce_window_shape = in TEST_F() local
4389 reduce_window_shape, pad, reduce_init_value, window, in TEST_F()
4406 EXPECT_TRUE(ShapeUtil::Equal(root->shape(), reduce_window_shape)) in TEST_F()
4408 << ShapeUtil::HumanString(reduce_window_shape); in TEST_F()
4469 const Shape reduce_window_shape = in TEST_F() local
4475 reduce_window_shape, convert, reduce_init_value, window, in TEST_F()
4492 EXPECT_TRUE(ShapeUtil::Equal(root->shape(), reduce_window_shape)) in TEST_F()
4494 << ShapeUtil::HumanString(reduce_window_shape); in TEST_F()
|