Home
last modified time | relevance | path

Searched defs:dimension0 (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dpad_test.cc48 auto dimension0 = r4_padding_on_dim0_dim1_.add_dimensions(); in PadTest() local
187 auto dimension0 = padding_config.add_dimensions(); in TEST_P() local
229 auto dimension0 = padding_config.add_dimensions(); in XLA_TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dshape_inference_test.cc2105 auto dimension0 = padding_config.add_dimensions(); in TEST_F() local
/aosp_15_r20/external/pytorch/benchmarks/static_runtime/
H A Dtest_generated_ops.cc7966 auto dimension0 = 1; in TEST() local