Searched refs:dims_size (Results 1 – 5 of 5) sorted by relevance
105 const int dims_size = dims->size; in GetTensorShape() local107 return RuntimeShape(dims_size, dims_data); in GetTensorShape()
57 size_t dims_size, size_t* bytes, size_t* type_size, in BytesRequired() argument64 *bytes = dims_size * (*type_size); in BytesRequired()
1477 num_dims = input_array.shape().dims_size(); in CreateOrCheckRnnStateArray()1548 QCHECK_EQ(specified_input_array.shape().dims_size(), in ResolveModelFlags()1549 dst_input_array->shape().dims_size()) in ResolveModelFlags()1552 << specified_input_array.shape().dims_size() in ResolveModelFlags()1555 << dst_input_array->shape().dims_size(); in ResolveModelFlags()1559 for (int i = 1; i < specified_input_array.shape().dims_size(); i++) { in ResolveModelFlags()1654 input_array_proto.shape().dims_size()); in ResolveModelFlags()
358 TfLiteStatus BytesRequired(TfLiteType type, const int* dims, size_t dims_size,
428 size_t dims_size, size_t* bytes) { in BytesRequired() argument433 for (int k = 0; k < dims_size; k++) count *= dims[k]; in BytesRequired()