Searched defs:new_num_elements (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
H A Dtensor_shape.cc244 int64 new_num_elements; local
246 new_num_elements = -1;
248 new_num_elements = MultiplyWithoutOverflow(num_elements(), size);
249 CHECK_LE(0, new_num_elements);
251 UnsafeAddDim(size, new_num_elements);
255 void TensorShapeBase<Shape>::UnsafeAddDim(int64 size, int64 new_num_elements) { argument
296 set_num_elements(new_num_elements);
497 int64 new_num_elements;
506 new_num_elements = -1;
508 new_num_elements
[all...]
H A Dtensor.h628 int64 new_num_elements = 1; local
630 new_num_elements *= new_sizes[d];
633 CHECK_EQ(new_num_elements, NumElements());
641 int64 new_num_elements = 1; local
643 new_num_elements *= new_sizes[d];
648 CHECK_EQ(new_num_elements * sizeof(T), NumElements() * element_size);
656 CHECK_EQ(new_num_elements, NumElements());
/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dliteral_test_util.cc759 int64 new_num_elements = 1; local
761 new_num_elements *= new_dimensions[i];
763 CHECK_EQ(ShapeUtil::ElementsIn(literal.shape()), new_num_elements); local

Completed in 295 milliseconds