Searched defs:b_3d (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | LinearAlgebraUtils.h | 215 auto b_3d = b.view({batchCount(b), b.size(-2), b.size(-1)}); in batch_iterator_with_broadcasting() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | array_elementwise_ops_test.cc | 2887 Array3D<float> b_3d({{{2.0f, 4.0f}, {6.0f, 8.0f}, {10.0f, 12.0f}}, variable
2993 Array3D<float> b_3d({{{7.0f, 1.0f}, {3.0f, 10.0f}, {15.0f, 6.0f}}}); in XLA_TEST_F() local
|