| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | fuse_binary_into_following_affine.cc | 52 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()
|
| D | resolve_constant_range.cc | 84 CHECK_EQ(RequiredBufferSizeForShape(start_array.shape()), 1) in Run()
86 CHECK_EQ(RequiredBufferSizeForShape(limit_array.shape()), 1) in Run()
88 CHECK_EQ(RequiredBufferSizeForShape(delta_array.shape()), 1) in Run()
|
| D | resolve_fake_quant_args_from_vars.cc | 56 CHECK_EQ(RequiredBufferSizeForShape(min_array.shape()), 1); in Run()
57 CHECK_EQ(RequiredBufferSizeForShape(max_array.shape()), 1); in Run()
|
| D | resolve_constant_pack.cc | 34 output_data.resize(RequiredBufferSizeForShape(output_array.shape())); in Pack()
42 int input_size = RequiredBufferSizeForShape(input_array.shape()); in Pack()
|
| D | resolve_constant_fill.cc | 35 data.resize(RequiredBufferSizeForShape(output_array.shape())); in ComputeFillArray()
78 CHECK_EQ(RequiredBufferSizeForShape(val_array.shape()), 1); in Run()
|
| D | resolve_strided_slice_attributes.cc | 31 buffer->data.resize(RequiredBufferSizeForShape(strided_slice_shape)); in PadAttributeArray()
64 if (toco::RequiredBufferSizeForShape(start_array.shape()) > 4) { in Run()
|
| D | resolve_batch_normalization.cc | 100 data.resize(RequiredBufferSizeForShape(mean_shape), data[0]); in Run()
105 const int buffer_size = RequiredBufferSizeForShape(param_shape); in Run()
|
| D | convert_expanddims_to_reshape.cc | 52 CHECK_EQ(RequiredBufferSizeForShape(axis_array.shape()), 1); in Run()
|
| D | propagate_fixed_sizes.cc | 360 CHECK_LE(RequiredBufferSizeForShape(dims_array.shape()), 4) in ProcessOpWithShapeInput()
387 const int input_overall_size = RequiredBufferSizeForShape(input_shape); in ProcessFullyConnectedOperator()
443 const int input_flat_size = RequiredBufferSizeForShape(input_shape); in ProcessTensorFlowReshapeOperator()
459 CHECK_EQ(input_flat_size, RequiredBufferSizeForShape(output_shape)) in ProcessTensorFlowReshapeOperator()
728 CHECK_EQ(RequiredBufferSizeForShape(start_array.shape()), 1) in ProcessRangeOperator()
730 CHECK_EQ(RequiredBufferSizeForShape(limit_array.shape()), 1) in ProcessRangeOperator()
732 CHECK_EQ(RequiredBufferSizeForShape(delta_array.shape()), 1) in ProcessRangeOperator()
769 CHECK_EQ(RequiredBufferSizeForShape(axis_array.shape()), 1) in ProcessTensorFlowSplitOperator()
839 CHECK_EQ(RequiredBufferSizeForShape(axis_array.shape()), 1) in ProcessTensorFlowSplitVOperator()
1826 CHECK_EQ(RequiredBufferSizeForShape(axis_array.shape()), 1) in ProcessArgMinMaxOperator()
[all …]
|
| D | resolve_constant_random_uniform.cc | 39 data.resize(RequiredBufferSizeForShape(output_array.shape())); in ComputeRandomUniformArray()
|
| D | resolve_constant_unary.cc | 211 const int output_buffer_size = RequiredBufferSizeForShape(output_shape); in Run()
217 const int input_buffer_size = RequiredBufferSizeForShape(input_shape); in Run()
|
| D | lstm_utils.cc | 50 buffer->data.resize(RequiredBufferSizeForShape(shape)); in CreateFloatArrayBuffer()
|
| D | resolve_reorder_axes.cc | 63 output_data.resize(RequiredBufferSizeForShape(output_array->shape())); in ReorderAxes()
|
| D | identify_relu1.cc | 43 RequiredBufferSizeForShape(op_array.shape()) != 1) { in CheckArrayIsScalarFloat()
|
| D | identify_l2_normalization.cc | 93 if (RequiredBufferSizeForShape(input_array.shape()) != 1) { in Run()
|
| D | resolve_multiply_by_zero.cc | 43 data.resize(RequiredBufferSizeForShape(array->shape())); in FillArrayWithZeros()
|
| D | resolve_constant_strided_slice.cc | 48 output_data.resize(RequiredBufferSizeForShape(output_array->shape())); in StridedSlice()
|
| D | resolve_constant_gather.cc | 40 output_data.resize(RequiredBufferSizeForShape(output_shape)); in Gather()
|
| D | resolve_constant_tile.cc | 76 output_data.resize(RequiredBufferSizeForShape(output_array->shape())); in Tile()
|
| D | resolve_constant_slice.cc | 38 output_data.resize(RequiredBufferSizeForShape(output_array->shape())); in Slice()
|
| D | resolve_constant_transpose.cc | 38 output_data.resize(RequiredBufferSizeForShape(output_shape)); in Transpose()
|
| D | unpartition_embedding_lookup.cc | 206 perm_data.resize(RequiredBufferSizeForShape(perm_array.shape())); in Run()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.h | 150 int RequiredBufferSizeForShape(const Shape& shape);
182 int source_buffer_size = RequiredBufferSizeForShape(source_array.shape());
183 int target_buffer_size = RequiredBufferSizeForShape(target_array->shape());
236 DCHECK_LT(index, RequiredBufferSizeForShape(shape));
|
| D | tooling_util.cc | 727 int RequiredBufferSizeForShape(const Shape& shape) { in RequiredBufferSizeForShape() function
1040 RequiredBufferSizeForShape(array->shape())); in CheckEachArray()
1866 param_array_data.resize(RequiredBufferSizeForShape(param_array.shape())); in CreateInt32Array()
1889 2 * RequiredBufferSizeForShape(weights_array.shape()); in EstimateArithmeticOpsCount()
1893 *result += RequiredBufferSizeForShape(output_array.shape()); in EstimateArithmeticOpsCount()
1904 *result = RequiredBufferSizeForShape(output_array.shape()); in EstimateArithmeticOpsCount()
1914 *result = num_adds * RequiredBufferSizeForShape(output_array.shape()); in EstimateArithmeticOpsCount()
1929 *result = 64 * RequiredBufferSizeForShape(output_array.shape()); in EstimateArithmeticOpsCount()
1938 *result = RequiredBufferSizeForShape(output_array.shape()) * in EstimateArithmeticOpsCount()
1948 *result = RequiredBufferSizeForShape(output_array.shape()) * in EstimateArithmeticOpsCount()
[all …]
|
| D | dump_graphviz.cc | 293 buffer_size = RequiredBufferSizeForShape(array.shape()); in GetArrayLabel()
506 buffer_size = RequiredBufferSizeForShape(array.shape()); in GetLog2BufferSize()
|