Searched refs:new_num_elements (Results 1 – 4 of 4) sorted by relevance
244 int64 new_num_elements; in AddDim() local246 new_num_elements = -1; in AddDim()248 new_num_elements = MultiplyWithoutOverflow(num_elements(), size); in AddDim()249 CHECK_LE(0, new_num_elements); in AddDim()251 UnsafeAddDim(size, new_num_elements); in AddDim()255 void TensorShapeBase<Shape>::UnsafeAddDim(int64 size, int64 new_num_elements) { in UnsafeAddDim() argument296 set_num_elements(new_num_elements); in UnsafeAddDim()497 int64 new_num_elements; in MakeShapeHelper() local506 new_num_elements = -1; in MakeShapeHelper()508 new_num_elements = -1; in MakeShapeHelper()[all …]
628 int64 new_num_elements = 1; in FillDimsAndValidateCompatibleShape() local630 new_num_elements *= new_sizes[d]; in FillDimsAndValidateCompatibleShape()633 CHECK_EQ(new_num_elements, NumElements()); in FillDimsAndValidateCompatibleShape()641 int64 new_num_elements = 1; in FillDimsAndValidateCompatibleShape() local643 new_num_elements *= new_sizes[d]; in FillDimsAndValidateCompatibleShape()648 CHECK_EQ(new_num_elements * sizeof(T), NumElements() * element_size); in FillDimsAndValidateCompatibleShape()656 CHECK_EQ(new_num_elements, NumElements()); in FillDimsAndValidateCompatibleShape()
267 void UnsafeAddDim(int64 size, int64 new_num_elements);
759 int64 new_num_elements = 1; in Reshape() local761 new_num_elements *= new_dimensions[i]; in Reshape()763 CHECK_EQ(ShapeUtil::ElementsIn(literal.shape()), new_num_elements); in Reshape()