Searched refs:num_shapes (Results 1 – 5 of 5) sorted by relevance
/external/libaom/test/ |
D | av1_nn_predict_test.cc | 61 const int num_shapes); 63 const int num_shapes, const int run_times); 191 const int num_shapes) { in RunNnPredictTest_all() argument 192 for (int i = 0; i < num_shapes; i++) RunNnPredictTest(&shapes[i]); in RunNnPredictTest_all() 196 const int num_shapes, in RunNnPredictSpeedTest_all() argument 198 for (int i = 0; i < num_shapes; i++) in RunNnPredictSpeedTest_all()
|
/external/tensorflow/tensorflow/c/ |
D | c_api_experimental.h | 274 int num_shapes);
|
D | c_api.h | 368 int num_shapes); 379 const void* const* protos, const size_t* proto_lens, int num_shapes, 643 int num_shapes, int64_t* storage, int storage_size, TF_Status* status);
|
D | c_api.cc | 917 int num_shapes) { in TF_SetAttrShapeList() argument 919 shapes.reserve(num_shapes); in TF_SetAttrShapeList() 920 for (int i = 0; i < num_shapes; ++i) { in TF_SetAttrShapeList() 954 const size_t* proto_lens, int num_shapes, in TF_SetAttrTensorShapeProtoList() argument 957 shapes.resize(num_shapes); in TF_SetAttrTensorShapeProtoList() 958 for (int i = 0; i < num_shapes; ++i) { in TF_SetAttrTensorShapeProtoList() 1416 int64_t** dims, int* num_dims, int num_shapes, in TF_OperationGetAttrShapeList() argument 1423 auto len = std::min(static_cast<int>(shapes.size()), num_shapes); in TF_OperationGetAttrShapeList()
|
/external/pytorch/test/torch_np/numpy_tests/core/ |
D | test_numeric.py | 1895 hynp.mutually_broadcastable_shapes(num_shapes=2, base_shape=arr.shape)
|