Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Darray_elementwise_ops_test.cc2095 Array3D<int32_t> a_3d({{{3, 9, -1}, {2, -10, 3}}, {{-2, 2, 8}, {12, 10, 4}}}); in XLA_TEST_F() local
2106 Array3D<int32_t> a_3d(2, 0, 3); in XLA_TEST_F() local
2883 Array3D<float> a_3d({{{1.0f, 2.0f}, {3.0f, 4.0f}, {5.0f, 6.0f}}, variable
2903 Array3D<float> a_3d({ in XLA_TEST_F() local
2927 Array3D<float> a_3d({ in XLA_TEST_F() local
2958 Array3D<float> a_3d({ in XLA_TEST_F() local
2989 Array3D<float> a_3d({{{1.0f, 2.0f}, {3.0f, 4.0f}, {5.0f, 6.0f}}, in XLA_TEST_F() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DLinearAlgebraUtils.h214 auto a_3d = a.view({batchCount(a), m, n}); in batch_iterator_with_broadcasting() local
H A DLinearAlgebra.cpp2710 const auto a_3d = a.view({-1, a.size(-2), a.size(-1)}); in mexp() local