Home
last modified time | relevance | path

Searched refs:RequiredBufferSizeForShape (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_range.cc85 CHECK_EQ(RequiredBufferSizeForShape(start_array.shape()), 1) in Run()
87 CHECK_EQ(RequiredBufferSizeForShape(limit_array.shape()), 1) in Run()
89 CHECK_EQ(RequiredBufferSizeForShape(delta_array.shape()), 1) in Run()
Dfuse_binary_into_following_affine.cc52 CHECK_EQ(RequiredBufferSizeForShape(operand.shape()), 1); in FuseAddOrSubParamsIntoFollowingAffine()
82 const int weights_size = RequiredBufferSizeForShape(weights_shape); in FuseAddOrSubParamsIntoFollowingAffine()
97 const int weights_size = RequiredBufferSizeForShape(weights_shape); in FuseAddOrSubParamsIntoFollowingAffine()
132 CHECK_EQ(RequiredBufferSizeForShape(operand.shape()), 1); in FuseMulOrDivParamsIntoFollowingAffine()
139 const int weights_size = RequiredBufferSizeForShape(weights.shape()); in FuseMulOrDivParamsIntoFollowingAffine()
Dresolve_fake_quant_args_from_vars.cc56 CHECK_EQ(RequiredBufferSizeForShape(min_array.shape()), 1); in Run()
57 CHECK_EQ(RequiredBufferSizeForShape(max_array.shape()), 1); in Run()
Dresolve_constant_pack.cc34 output_data.resize(RequiredBufferSizeForShape(output_array.shape())); in Pack()
42 int input_size = RequiredBufferSizeForShape(input_array.shape()); in Pack()
Dresolve_constant_fill.cc35 data.resize(RequiredBufferSizeForShape(output_array.shape())); in ComputeFillArray()
78 CHECK_EQ(RequiredBufferSizeForShape(val_array.shape()), 1); in Run()
Dresolve_strided_slice_attributes.cc31 buffer->data.resize(RequiredBufferSizeForShape(strided_slice_shape)); in PadAttributeArray()
64 if (toco::RequiredBufferSizeForShape(start_array.shape()) > 4) { in Run()
Dresolve_batch_normalization.cc102 data.resize(RequiredBufferSizeForShape(mean_shape), data[0]); in Run()
107 const int buffer_size = RequiredBufferSizeForShape(param_shape); in Run()
Didentify_util.cc33 RequiredBufferSizeForShape(op_array.shape()) != 1) { in CheckArrayIsScalarFloat()
Dresolve_constant_random_uniform.cc39 data.resize(RequiredBufferSizeForShape(output_array.shape())); in ComputeRandomUniformArray()
Dconvert_expanddims_to_reshape.cc52 CHECK_EQ(RequiredBufferSizeForShape(axis_array.shape()), 1); in Run()
Dlstm_utils.cc50 buffer->data.resize(RequiredBufferSizeForShape(shape)); in CreateFloatArrayBuffer()
Dpropagate_fixed_sizes.cc361 CHECK_LE(RequiredBufferSizeForShape(dims_array.shape()), 6) in ProcessOpWithShapeInput()
388 const int input_overall_size = RequiredBufferSizeForShape(input_shape); in ProcessFullyConnectedOperator()
444 const int input_flat_size = RequiredBufferSizeForShape(input_shape); in ProcessTensorFlowReshapeOperator()
460 CHECK_EQ(input_flat_size, RequiredBufferSizeForShape(output_shape)) in ProcessTensorFlowReshapeOperator()
733 CHECK_EQ(RequiredBufferSizeForShape(start_array.shape()), 1) in ProcessRangeOperator()
735 CHECK_EQ(RequiredBufferSizeForShape(limit_array.shape()), 1) in ProcessRangeOperator()
737 CHECK_EQ(RequiredBufferSizeForShape(delta_array.shape()), 1) in ProcessRangeOperator()
782 CHECK_EQ(RequiredBufferSizeForShape(axis_array.shape()), 1) in ProcessTensorFlowSplitOperator()
852 CHECK_EQ(RequiredBufferSizeForShape(axis_array.shape()), 1) in ProcessTensorFlowSplitVOperator()
1846 CHECK_EQ(RequiredBufferSizeForShape(axis_array.shape()), 1) in ProcessArgMinMaxOperator()
[all …]
Dresolve_reorder_axes.cc63 output_data.resize(RequiredBufferSizeForShape(output_array->shape())); in ReorderAxes()
Didentify_l2_normalization.cc79 if (RequiredBufferSizeForShape(input_array.shape()) != 1) { in Run()
Dresolve_constant_unary.cc212 const int output_buffer_size = RequiredBufferSizeForShape(output_shape); in Run()
218 const int input_buffer_size = RequiredBufferSizeForShape(input_shape); in Run()
Dresolve_multiply_by_zero.cc43 data.resize(RequiredBufferSizeForShape(array->shape())); in FillArrayWithZeros()
Dresolve_constant_gather.cc40 output_data.resize(RequiredBufferSizeForShape(output_shape)); in Gather()
Dresolve_constant_strided_slice.cc48 output_data.resize(RequiredBufferSizeForShape(output_array->shape())); in StridedSlice()
Dresolve_constant_slice.cc38 output_data.resize(RequiredBufferSizeForShape(output_array->shape())); in Slice()
Dresolve_constant_tile.cc77 output_data.resize(RequiredBufferSizeForShape(output_array->shape())); in Tile()
Dresolve_constant_transpose.cc38 output_data.resize(RequiredBufferSizeForShape(output_shape)); in Transpose()
Dunpartition_embedding_lookup.cc208 perm_data.resize(RequiredBufferSizeForShape(perm_array.shape())); in Run()
Dresolve_constant_binary.cc93 const int output_buffer_size = RequiredBufferSizeForShape(output_shape); in EvaluateBinaryOperatorOnConstantInputs()
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.h147 int RequiredBufferSizeForShape(const Shape& shape);
179 int source_buffer_size = RequiredBufferSizeForShape(source_array.shape());
180 int target_buffer_size = RequiredBufferSizeForShape(target_array->shape());
233 DCHECK_LT(index, RequiredBufferSizeForShape(shape));
Dtooling_util.cc755 int RequiredBufferSizeForShape(const Shape& shape) { in RequiredBufferSizeForShape() function
1070 RequiredBufferSizeForShape(array->shape())) in CheckEachArray()
1890 param_array_data.resize(RequiredBufferSizeForShape(param_array.shape())); in CreateInt32Array()
1913 2 * RequiredBufferSizeForShape(weights_array.shape()); in EstimateArithmeticOpsCount()
1917 *result += RequiredBufferSizeForShape(output_array.shape()); in EstimateArithmeticOpsCount()
1948 *result = RequiredBufferSizeForShape(output_array.shape()); in EstimateArithmeticOpsCount()
1958 *result = num_adds * RequiredBufferSizeForShape(output_array.shape()); in EstimateArithmeticOpsCount()
1973 *result = 64 * RequiredBufferSizeForShape(output_array.shape()); in EstimateArithmeticOpsCount()
1982 *result = RequiredBufferSizeForShape(output_array.shape()) * in EstimateArithmeticOpsCount()
1992 *result = RequiredBufferSizeForShape(output_array.shape()) * in EstimateArithmeticOpsCount()
[all …]

12