Searched refs:set_num_elements (Results 1 – 5 of 5) sorted by relevance
148 set_num_elements(-1); in TensorShapeBase()151 set_num_elements(1); in TensorShapeBase()168 out->set_num_elements(-1); in BuildTensorShapeBase()171 out->set_num_elements(1); in BuildTensorShapeBase()246 set_num_elements(neg ? -1 : size); in InitDims()255 set_num_elements(neg ? -1 : (size0 * size1)); in InitDims()266 set_num_elements(neg ? -1 : (size0 * size1 * size2)); in InitDims()279 set_num_elements(neg ? -1 : (size0 * size1 * size2 * size3)); in InitDims()286 set_num_elements(1); in InitDims()304 set_num_elements(-1); in TensorShapeBase()[all …]
144 void set_num_elements(int64 n) { num_elements_ = n; } in set_num_elements() function709 set_num_elements(0); in TensorShapeBase()
1497 node_proto->set_num_elements(num_elements_); in ToProto()
174 metadata.set_num_elements(num_elements); in WriteMetadataFile()
377 layer.set_num_elements(5)