Searched refs:new_num_elements (Results 1 – 7 of 7) sorted by relevance
| /external/mesa3d/src/gallium/auxiliary/util/ |
| D | u_idalloc.c | 53 util_idalloc_resize(struct util_idalloc *buf, unsigned new_num_elements) in util_idalloc_resize() argument
55 new_num_elements = align(new_num_elements, 32); in util_idalloc_resize()
57 if (new_num_elements > buf->num_elements) { in util_idalloc_resize()
61 (new_num_elements / 32) * sizeof(*buf->data)); in util_idalloc_resize()
63 for (i = buf->num_elements / 32; i < new_num_elements / 32; i++) in util_idalloc_resize()
65 buf->num_elements = new_num_elements; in util_idalloc_resize()
|
| D | u_idalloc.h | 50 util_idalloc_resize(struct util_idalloc *buf, unsigned new_num_elements);
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_shape.cc | 327 int64 new_num_elements; in AddDim() local
329 new_num_elements = -1; in AddDim()
331 new_num_elements = MultiplyWithoutOverflow(num_elements(), size); in AddDim()
332 CHECK_LE(0, new_num_elements); in AddDim()
334 UnsafeAddDim(size, new_num_elements); in AddDim()
338 void TensorShapeBase<Shape>::UnsafeAddDim(int64 size, int64 new_num_elements) { in UnsafeAddDim() argument
379 set_num_elements(new_num_elements); in UnsafeAddDim()
580 int64 new_num_elements; in MakeShapeHelper() local
589 new_num_elements = -1; in MakeShapeHelper()
591 new_num_elements = -1; in MakeShapeHelper()
[all …]
|
| D | tensor.h | 779 int64 new_num_elements = 1; in FillDimsAndValidateCompatibleShape() local
781 new_num_elements *= new_sizes[d]; in FillDimsAndValidateCompatibleShape()
784 CHECK_EQ(new_num_elements, NumElements()); in FillDimsAndValidateCompatibleShape()
792 int64 new_num_elements = 1; in FillDimsAndValidateCompatibleShape() local
794 new_num_elements *= new_sizes[d]; in FillDimsAndValidateCompatibleShape()
799 CHECK_EQ(new_num_elements * sizeof(T), NumElements() * element_size); in FillDimsAndValidateCompatibleShape()
807 CHECK_EQ(new_num_elements, NumElements()); in FillDimsAndValidateCompatibleShape()
|
| D | tensor_shape.h | 268 void UnsafeAddDim(int64 size, int64 new_num_elements);
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_util.cc | 295 int64 new_num_elements = 1; in ReshapeSlice() local
297 new_num_elements *= new_dimensions[i]; in ReshapeSlice()
299 CHECK_EQ(ShapeUtil::ElementsIn(literal.shape()), new_num_elements); in ReshapeSlice()
|
| /external/mesa3d/src/gallium/drivers/radeonsi/ |
| D | si_descriptors.c | 2160 unsigned new_num_elements = desc->num_elements * 2; in si_get_first_free_bindless_slot() local
2163 new_num_elements * slot_size); in si_get_first_free_bindless_slot()
2164 desc->num_elements = new_num_elements; in si_get_first_free_bindless_slot()
2165 desc->num_active_slots = new_num_elements; in si_get_first_free_bindless_slot()
|