Searched refs:reduce_window_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 255 auto reduce_window_shape = new_convolution->shape(); in HandleBatchGroupCount() local
256 reduce_window_shape.set_dimensions(output_batch_dimension, 1); in HandleBatchGroupCount()
261 reduce_window_shape.set_element_type(F32); in HandleBatchGroupCount()
301 reduce_window_shape, new_filter, zero_scalar, window, in HandleBatchGroupCount()
|
| D | algebraic_simplifier_test.cc | 3615 const Shape reduce_window_shape = in TEST_F() local
3621 reduce_window_shape, pad, reduce_init_value, window, in TEST_F()
3638 EXPECT_TRUE(ShapeUtil::Equal(root->shape(), reduce_window_shape)) in TEST_F()
3640 << ShapeUtil::HumanString(reduce_window_shape); in TEST_F()
3701 const Shape reduce_window_shape = in TEST_F() local
3707 reduce_window_shape, convert, reduce_init_value, window, in TEST_F()
3724 EXPECT_TRUE(ShapeUtil::Equal(root->shape(), reduce_window_shape)) in TEST_F()
3726 << ShapeUtil::HumanString(reduce_window_shape); in TEST_F()
|