Searched defs:shape_w (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | FuseBatchNormalizationFixture.h | 51 …void setup(TensorShape shape_w, DataType data_type, DataLayout data_layout, bool in_place, bool wi… in setup()
64 …std::pair<TensorType, TensorType> compute_target(TensorShape shape_w, DataType data_type, DataLayo… in compute_target()
141 …std::pair<SimpleTensor<T>, SimpleTensor<T>> compute_reference(TensorShape shape_w, DataType data_t… in compute_reference()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_mps.py | 6593 def helper1(shape_x, shape_y, shape_z, shape_w): argument
|