Searched defs:shape_v (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/include/libnpy/ |
| H A D | npy.hpp | 390 auto shape_v = npy::pyparse::parse_tuple(shape_s); in parse_header() local
502 std::vector <ndarray_len_t> shape_v(shape, shape + n_dims); in SaveArrayAsNumpy() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | ctc_ops.cc | 166 ShapeHandle shape_v = c->Vector(2); in __anon8bbafe030402() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_sparse.py | 1682 def _test_spadd_shape(nnz, shape_i, shape_v=None): argument
1858 def _test_basic_ops_shape(self, nnz_x1, nnz_x2, shape_i, shape_v, dtype, device, coalesced): argument
1991 def _test_sparse_mask_shape(self, nnz_x1, nnz_x2, shape_i, shape_v, dtype, device, coalesced): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | shape_refiner_test.cc | 294 Node* shape_v; in TEST_F() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | shape_type_inference.cpp | 1453 c10::SymbolicShape shape_v(dims); in ComputeConstant() local
|