Home
last modified time | relevance | path

Searched defs:array4d (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dround_trip_transfer_test.cc135 Array4D<float> array4d(2, 2, 256, 256); in TEST_F() local
H A Darray_elementwise_ops_test.cc2139 auto array4d = ConstantR4FromArray4D<float>( in XLA_TEST_F() local
2155 auto array4d = ConstantR4FromArray4D<float>(&builder, arg); in XLA_TEST_F() local