Searched refs:new_num_elements (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_shape.cc | 399 int64_t new_num_elements; in AddDim() local
401 new_num_elements = -1; in AddDim()
403 new_num_elements = MultiplyWithoutOverflow(num_elements(), size); in AddDim()
404 CHECK_LE(0, new_num_elements); in AddDim()
406 UnsafeAddDim(size, new_num_elements); in AddDim()
426 int64_t new_num_elements; in AddDimWithStatus() local
428 new_num_elements = -1; in AddDimWithStatus()
430 new_num_elements = MultiplyWithoutOverflow(num_elements(), size); in AddDimWithStatus()
431 if (TF_PREDICT_FALSE(new_num_elements < 0)) { in AddDimWithStatus()
434 ", result: ", new_num_elements); in AddDimWithStatus()
[all …]
|
| D | tensor.h | 827 int64_t new_num_elements = 1; in FillDimsAndValidateCompatibleShape() local
829 new_num_elements *= new_sizes[d]; in FillDimsAndValidateCompatibleShape()
832 CHECK_EQ(new_num_elements, NumElements()); in FillDimsAndValidateCompatibleShape()
840 int64_t new_num_elements = 1; in FillDimsAndValidateCompatibleShape() local
842 new_num_elements *= new_sizes[d]; in FillDimsAndValidateCompatibleShape()
847 CHECK_EQ(new_num_elements * static_cast<int64_t>(sizeof(T)), in FillDimsAndValidateCompatibleShape()
856 CHECK_EQ(new_num_elements, NumElements()); in FillDimsAndValidateCompatibleShape()
|
| D | tensor_shape.h | 336 void UnsafeAddDim(int64_t size, int64_t new_num_elements);
|
| /external/tensorflow/tensorflow/security/advisory/ |
| D | tfsa-2021-071.md | 72 int64 new_num_elements;
74 new_num_elements = -1;
76 new_num_elements = MultiplyWithoutOverflow(num_elements(), size);
77 if (TF_PREDICT_FALSE(new_num_elements < 0)) {
80 ", result: ", new_num_elements);
|
| D | tfsa-2021-012.md | 67 int64 new_num_elements;
69 new_num_elements = -1;
71 new_num_elements = MultiplyWithoutOverflow(num_elements(), size);
72 if (TF_PREDICT_FALSE(new_num_elements < 0)) {
75 ", result: ", new_num_elements);
|
| D | tfsa-2021-024.md | 71 int64 new_num_elements;
73 new_num_elements = -1;
75 new_num_elements = MultiplyWithoutOverflow(num_elements(), size);
76 if (TF_PREDICT_FALSE(new_num_elements < 0)) {
79 ", result: ", new_num_elements);
|
| /external/mesa3d/src/util/ |
| D | u_idalloc.c | 47 util_idalloc_resize(struct util_idalloc *buf, unsigned new_num_elements) in util_idalloc_resize() argument
49 if (new_num_elements > buf->num_elements) { in util_idalloc_resize()
50 buf->data = realloc(buf->data, new_num_elements * sizeof(*buf->data)); in util_idalloc_resize()
52 (new_num_elements - buf->num_elements) * sizeof(*buf->data)); in util_idalloc_resize()
53 buf->num_elements = new_num_elements; in util_idalloc_resize()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_concat_op.cc | 129 int64_t new_num_elements = 1; in Compute() local
136 new_num_elements = in Compute()
137 MultiplyWithoutOverflow(new_num_elements, input_shape_vector(j)); in Compute()
138 if (new_num_elements < 0) { in Compute()
|
| /external/icing/icing/file/ |
| D | file-backed-vector.h | 359 libtextclassifier3::Status TruncateTo(int32_t new_num_elements);
949 int32_t new_num_elements) { in TruncateTo() argument
950 if (new_num_elements < 0) { in TruncateTo()
952 "Truncated length %d must be >= 0", new_num_elements)); in TruncateTo()
955 if (new_num_elements >= header()->num_elements) { in TruncateTo()
958 new_num_elements, header()->num_elements)); in TruncateTo()
968 header()->num_elements = new_num_elements; in TruncateTo()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_util.cc | 391 int64_t new_num_elements = 1; in ReshapeSlice() local
393 new_num_elements *= new_dimensions[i]; in ReshapeSlice()
395 CHECK_EQ(ShapeUtil::ElementsIn(literal.shape()), new_num_elements); in ReshapeSlice()
|
| /external/mesa3d/src/gallium/drivers/radeonsi/ |
| D | si_descriptors.c | 2515 unsigned new_num_elements = desc->num_elements * 2; in si_get_first_free_bindless_slot() local
2518 REALLOC(desc->list, desc->num_elements * slot_size, new_num_elements * slot_size); in si_get_first_free_bindless_slot()
2519 desc->num_elements = new_num_elements; in si_get_first_free_bindless_slot()
2520 desc->num_active_slots = new_num_elements; in si_get_first_free_bindless_slot()
|