Home
last modified time | relevance | path

Searched refs:MultiplyWithoutOverflow (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Doverflow_test.cc65 int64_t xy = MultiplyWithoutOverflow(x, y); in TEST()
78 EXPECT_LT(MultiplyWithoutOverflow(n, 0), 0) << n; in TEST()
79 EXPECT_LT(MultiplyWithoutOverflow(0, n), 0) << n; in TEST()
80 EXPECT_LT(MultiplyWithoutOverflow(n, n), 0) << n; in TEST()
Doverflow.h27 inline int64_t MultiplyWithoutOverflow(const int64_t x, const int64_t y) { in MultiplyWithoutOverflow() function
/external/tensorflow/tensorflow/core/framework/
Dtensor_shape.cc69 num_elements = MultiplyWithoutOverflow(num_elements, d); in IsValid()
89 num_elements = MultiplyWithoutOverflow(num_elements, d.size()); in IsValid()
127 num_elements = MultiplyWithoutOverflow(num_elements, d.size()); in IsValidShape()
383 n = MultiplyWithoutOverflow(n, dim.size); in RecomputeNumElements()
403 new_num_elements = MultiplyWithoutOverflow(num_elements(), size); in AddDim()
430 new_num_elements = MultiplyWithoutOverflow(num_elements(), size); in AddDimWithStatus()
820 new_num_elements = MultiplyWithoutOverflow(out->num_elements(), dim); in MakeShapeHelper()
1016 n = MultiplyWithoutOverflow(n, dim); in NumElements()
Dshape_inference.cc1110 const int64_t product = MultiplyWithoutOverflow(first_value, second_value); in Multiply()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dspacetobatch_op.cc71 block_num_elems = MultiplyWithoutOverflow(block_num_elems, block_shape[i]); in SpaceToBatch()
82 MultiplyWithoutOverflow(batch_size, block_num_elems); in SpaceToBatch()
/external/tensorflow/tensorflow/core/kernels/
Dspacetobatch_op.cc108 MultiplyWithoutOverflow(block_shape_product, block_shape[block_dim]); in SpaceToBatchOpCompute()
140 const int64_t output_shape = MultiplyWithoutOverflow( in SpaceToBatchOpCompute()
Dreshape_op.h146 if (MultiplyWithoutOverflow(shape->num_elements(), size) < 0) { in ValidateSizes()
Done_hot_op.cc85 MultiplyWithoutOverflow(indices_shape.num_elements(), depth_v) >= 0, in Compute()
Dsparse_concat_op.cc137 MultiplyWithoutOverflow(new_num_elements, input_shape_vector(j)); in Compute()
Dfractional_avg_pool_op.cc251 num_elements = MultiplyWithoutOverflow( in Compute()
/external/tensorflow/tensorflow/compiler/xla/
Doverflow_util.h28 inline int64_t MultiplyWithoutOverflow(const int64_t x, const int64_t y) { in MultiplyWithoutOverflow() function
Dshape_util.cc234 dense_shape_size = MultiplyWithoutOverflow(dense_shape_size, d); in FillNewShape()
926 dense_shape_size = MultiplyWithoutOverflow(dense_shape_size, dim); in ValidateShapeSize()
931 dense_shape_size = MultiplyWithoutOverflow( in ValidateShapeSize()
/external/tensorflow/tensorflow/security/advisory/
Dtfsa-2021-198.md7 …n a tensor must fit within an `int64_t`. If an overflow occurs, `MultiplyWithoutOverflow` would re…
Dtfsa-2021-071.md76 new_num_elements = MultiplyWithoutOverflow(num_elements(), size);
Dtfsa-2021-012.md71 new_num_elements = MultiplyWithoutOverflow(num_elements(), size);
Dtfsa-2021-024.md75 new_num_elements = MultiplyWithoutOverflow(num_elements(), size);
/external/tensorflow/tensorflow/core/grappler/costs/
Danalytical_cost_estimator.cc103 size = MultiplyWithoutOverflow(size, std::max<int64_t>(1, dim.size())); in AddCostNode()
Dop_level_cost_estimator.cc1548 int64_t new_tensor_size = MultiplyWithoutOverflow(tensor_size, dim.size()); in CalculateTensorElementCount()
1565 int64_t tensor_size = MultiplyWithoutOverflow(count, size); in CalculateTensorSize()
1625 MultiplyWithoutOverflow(output_size, dim.size()); in CalculateOutputSize()
2696 int64_t crop_area = MultiplyWithoutOverflow(crop_height, crop_width); in PredictCropAndResize()
2701 int64_t crop_volume = MultiplyWithoutOverflow(crop_area, num_boxes); in PredictCropAndResize()
2706 int64_t crop_depth = MultiplyWithoutOverflow(crop_height, num_boxes); in PredictCropAndResize()
Dutils.cc221 int64_t tensor_size = MultiplyWithoutOverflow(num_elems, size); in CalculateTensorSize()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.cc450 num_elements = MultiplyWithoutOverflow(num_elements, *i); in ConstantInputAsShape()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding.cc685 MultiplyWithoutOverflow(product_of_dimensions, dimension); in FromProto()