HomeSort by relevance Sort by last modified time
    Searched defs:new_num_elements (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/
literal_util.cc 295 int64 new_num_elements = 1; local
297 new_num_elements *= new_dimensions[i];
299 CHECK_EQ(ShapeUtil::ElementsIn(literal.shape()), new_num_elements); local
  /external/tensorflow/tensorflow/core/framework/
tensor_shape.cc 327 int64 new_num_elements; local
329 new_num_elements = -1;
331 new_num_elements = MultiplyWithoutOverflow(num_elements(), size);
332 CHECK_LE(0, new_num_elements);
334 UnsafeAddDim(size, new_num_elements);
338 void TensorShapeBase<Shape>::UnsafeAddDim(int64 size, int64 new_num_elements) {
379 set_num_elements(new_num_elements);
580 int64 new_num_elements;
589 new_num_elements = -1;
591 new_num_elements = -1
    [all...]
tensor.h 779 int64 new_num_elements = 1; local
781 new_num_elements *= new_sizes[d];
784 CHECK_EQ(new_num_elements, NumElements());
792 int64 new_num_elements = 1; local
794 new_num_elements *= new_sizes[d];
799 CHECK_EQ(new_num_elements * sizeof(T), NumElements() * element_size);
807 CHECK_EQ(new_num_elements, NumElements());
    [all...]
  /external/mesa3d/src/gallium/drivers/radeonsi/
si_descriptors.c 2160 unsigned new_num_elements = desc->num_elements * 2; local
    [all...]

Completed in 1250 milliseconds