Home
last modified time | relevance | path

Searched refs:dimension0 (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dpad_test.cc50 auto dimension0 = r4_padding_on_dim0_dim1_.add_dimensions(); in PadTest() local
51 dimension0->set_edge_padding_low(1); in PadTest()
52 dimension0->set_edge_padding_high(0); in PadTest()
53 dimension0->set_interior_padding(2); in PadTest()
189 auto dimension0 = padding_config.add_dimensions(); in TEST_P() local
190 dimension0->set_edge_padding_low(0); in TEST_P()
191 dimension0->set_edge_padding_high(0); in TEST_P()
192 dimension0->set_interior_padding(0); in TEST_P()
231 auto dimension0 = padding_config.add_dimensions(); in XLA_TEST_P() local
232 dimension0->set_edge_padding_low(0); in XLA_TEST_P()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc1424 auto dimension0 = padding_config.add_dimensions(); in TEST_F() local
1425 dimension0->set_edge_padding_low(0); in TEST_F()
1426 dimension0->set_edge_padding_high(2); in TEST_F()
1427 dimension0->set_interior_padding(3); in TEST_F()