Searched refs:num_shapes (Results 1 – 4 of 4) sorted by relevance
| /external/libaom/libaom/test/ |
| D | av1_nn_predict_test.cc | 60 const int num_shapes);
62 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.h | 576 int num_shapes);
587 const void* const* protos, const size_t* proto_lens, int num_shapes,
846 int num_shapes, int64_t* storage, int storage_size, TF_Status* status);
|
| D | c_api.cc | 1346 int num_shapes) { in TF_SetAttrShapeList() argument
1348 shapes.reserve(num_shapes); in TF_SetAttrShapeList()
1349 for (int i = 0; i < num_shapes; ++i) { in TF_SetAttrShapeList()
1385 const size_t* proto_lens, int num_shapes, in TF_SetAttrTensorShapeProtoList() argument
1388 shapes.resize(num_shapes); in TF_SetAttrTensorShapeProtoList()
1389 for (int i = 0; i < num_shapes; ++i) { in TF_SetAttrTensorShapeProtoList()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 3109 int num_shapes = ConstructInputReducedShapeAndCastInputIrArrayToShape( in EmitKernel() local
3112 DCHECK_EQ(num_shapes, num_params); in EmitKernel()
|