Home
last modified time | relevance | path

Searched refs:dimensions_count (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc90 int rank_x = input_shape_x.dimensions_count(); in ComputeBinaryOperatorOutputSize()
91 int rank_y = input_shape_y.dimensions_count(); in ComputeBinaryOperatorOutputSize()
131 CHECK(input_shape.dimensions_count() == 4) in ProcessConvOperator()
133 << "\" is " << input_shape.dimensions_count() << "D."; in ProcessConvOperator()
141 CHECK_EQ(weights_shape.dimensions_count(), 4); in ProcessConvOperator()
152 CHECK_EQ(output_array.shape().dimensions_count(), 4); in ProcessConvOperator()
187 CHECK(specified_output_shape_array.shape().dimensions_count() == 1 && in ProcessTransposeConvOperator()
203 CHECK_EQ(weights_shape.dimensions_count(), 4) in ProcessTransposeConvOperator()
230 CHECK_EQ(input_shape.dimensions_count(), 4) in ProcessTransposeConvOperator()
260 CHECK_EQ(input_shape.dimensions_count(), 4); in ProcessDepthwiseConvOperator()
[all …]
Dfuse_binary_into_preceding_affine.cc52 const int depth = bias_shape.dims(bias_shape.dimensions_count() - 1); in FuseAddOrSubParamsIntoPrecedingAffine()
54 if (operand_shape.dimensions_count() >= 1 && in FuseAddOrSubParamsIntoPrecedingAffine()
55 operand_shape.dims(operand_shape.dimensions_count() - 1) == in FuseAddOrSubParamsIntoPrecedingAffine()
56 bias_shape.dims(bias_shape.dimensions_count() - 1)) { in FuseAddOrSubParamsIntoPrecedingAffine()
58 } else if (operand_shape.dimensions_count() == 0 || in FuseAddOrSubParamsIntoPrecedingAffine()
59 operand_shape.dims(operand_shape.dimensions_count() - 1) == 1) { in FuseAddOrSubParamsIntoPrecedingAffine()
125 if (operand_shape.dimensions_count() >= 1 && in FuseMulOrDivParamsIntoPrecedingAffine()
126 operand_shape.dims(operand_shape.dimensions_count() - 1) == in FuseMulOrDivParamsIntoPrecedingAffine()
127 bias_shape.dims(bias_shape.dimensions_count() - 1)) { in FuseMulOrDivParamsIntoPrecedingAffine()
129 } else if (operand_shape.dimensions_count() == 0 || in FuseMulOrDivParamsIntoPrecedingAffine()
[all …]
Dresolve_constant_unary.cc56 std::vector<int> reduction_mask(input_shape.dimensions_count(), 1); in ReduceGeneric()
59 CHECK_LT(axis, input_shape.dimensions_count()); in ReduceGeneric()
64 std::vector<int> output_indices(input_shape.dimensions_count()); in ReduceGeneric()
69 for (int i = 0; i < input_shape.dimensions_count(); ++i) { in ReduceGeneric()
86 for (int i = 0; i < output_shape.dimensions_count(); ++i) { in ReduceGeneric()
210 const int output_dims_count = output_shape.dimensions_count(); in Run()
Dconvert_squeeze_to_reshape.cc50 if (input_array.shape().dimensions_count() == 0) { in Run()
64 if (output_shape.dimensions_count() == 0) { in Run()
Dunroll_batch_matmul.cc41 const int dims_count = input_array_a.shape().dimensions_count(); in UnrollBatchMatMul3D()
165 const int32 dims = input_array.shape().dimensions_count(); in GetTransposePerm()
177 const int32 dims = input_shape.dimensions_count(); in GetTransposeShape()
249 const int dims = input_array_a.shape().dimensions_count(); in Run()
275 CHECK_GE(input_array_a.shape().dimensions_count(), 3) in Run()
Dconvert_trivial_pack_to_reshape.cc48 if (input_array.shape().dimensions_count() == 0) { in Run()
66 1 + input_array.shape().dimensions_count()}; in Run()
Dresolve_constant_transpose.cc40 CHECK(input_shape.dimensions_count() == output_shape.dimensions_count()); in Transpose()
41 const int dim = input_shape.dimensions_count(); in Transpose()
Dresolve_constant_binary.cc94 const int dims_count = output_shape.dimensions_count(); in EvaluateBinaryOperatorOnConstantInputs()
106 CHECK_EQ(input0_shape.dimensions_count(), input1_shape.dimensions_count()); in EvaluateBinaryOperatorOnConstantInputs()
107 CHECK_EQ(input0_shape.dimensions_count(), dims_count); in EvaluateBinaryOperatorOnConstantInputs()
Dremove_trivial_binary.cc92 if (input_array_0.shape().dimensions_count() == in Run()
93 input_array_1.shape().dimensions_count() && in Run()
Dshuffle_fc_weights.cc60 for (int i = 1; i < input_shape.dimensions_count() - 1; i++) { in Run()
92 if (weights_shape.dimensions_count() != 2) { in Run()
Dmove_binary_operator_before_reshape.cc29 int shape_end = shape.dimensions_count() - 1; in IsTailOfShape()
30 int tail_end = tail.dimensions_count() - 1; in IsTailOfShape()
Dresolve_constant_concatenation.cc83 for (int i = 0; i < concatenated_array->shape().dimensions_count(); i++) { in ConcatenateTensorBuffers()
93 for (int i = concatenation_axis; i < array_shape.dimensions_count(); i++) { in ConcatenateTensorBuffers()
Dunpartition_embedding_lookup.cc199 partition_array.shape().dimensions_count(); in Run()
218 merged_gather_op->input_rank = partition_array.shape().dimensions_count(); in Run()
Dremove_trivial_concatenation_input.cc49 input_array.has_shape() && input_array.shape().dimensions_count() == 0; in Run()
Dresolve_constant_shape_or_rank.cc59 output_buffer.data[0] = input_array.shape().dimensions_count(); in Run()
Dfuse_binary_into_following_affine.cc81 CHECK_EQ(output_depth, bias_shape.dims(bias_shape.dimensions_count() - 1)); in FuseAddOrSubParamsIntoFollowingAffine()
96 weights_shape.dims(weights_shape.dimensions_count() - 1); in FuseAddOrSubParamsIntoFollowingAffine()
Dresolve_strided_slice_attributes.cc82 int num_input_axes = input_array.shape().dimensions_count(); in Run()
Dconvert_reorder_axes.cc39 CHECK_EQ(input_shape.dimensions_count(), 4); in CreateReshapeFromReorderAxes()
Dresolve_constant_gather.cc45 for (int i = 1; i < input_shape.dimensions_count(); ++i) { in Gather()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h131 explicit RuntimeShape(int dimensions_count) : size_(dimensions_count) { in RuntimeShape() argument
132 if (dimensions_count > kMaxSmallSize) { in RuntimeShape()
136 dims_pointer_ = new int32[dimensions_count]; in RuntimeShape()
148 RuntimeShape(int dimensions_count, const int32* dims_data) : size_(0) { in RuntimeShape() argument
149 ReplaceWith(dimensions_count, dims_data); in RuntimeShape()
205 inline void Resize(int dimensions_count) { in Resize() argument
213 size_ = dimensions_count; in Resize()
214 if (dimensions_count > kMaxSmallSize) { in Resize()
218 dims_pointer_ = new int32[dimensions_count]; in Resize()
223 inline void ReplaceWith(int dimensions_count, const int32* dims_data) { in ReplaceWith() argument
[all …]
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dtypes.h132 explicit RuntimeShape(int dimensions_count) : size_(dimensions_count) { in RuntimeShape() argument
133 if (dimensions_count > kMaxSmallSize) { in RuntimeShape()
134 dims_pointer_ = new int32[dimensions_count]; in RuntimeShape()
145 RuntimeShape(int dimensions_count, const int32* dims_data) : size_(0) { in RuntimeShape() argument
146 ReplaceWith(dimensions_count, dims_data); in RuntimeShape()
198 inline void Resize(int dimensions_count) { in Resize() argument
202 size_ = dimensions_count; in Resize()
203 if (dimensions_count > kMaxSmallSize) { in Resize()
204 dims_pointer_ = new int32[dimensions_count]; in Resize()
208 inline void ReplaceWith(int dimensions_count, const int32* dims_data) { in ReplaceWith() argument
[all …]
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc498 if (array_shape.dimensions_count() == 0) { in LogArray()
611 CHECK_GE(new_shape_size, shape->dimensions_count()); in ExtendShape()
612 const int size_increase = new_shape_size - shape->dimensions_count(); in ExtendShape()
619 CHECK_LE(new_shape_size, shape->dimensions_count()); in UnextendShape()
620 const int size_reduction = shape->dimensions_count() - new_shape_size; in UnextendShape()
648 for (int i = 0; i < shape.dimensions_count(); ++i) { in IsNonEmpty()
655 for (int i = 0; i < shape.dimensions_count(); ++i) { in CheckNonEmptyShapeDimensions()
667 if (shape1.dimensions_count() > shape0.dimensions_count()) { in ShapesAgreeUpToBroadcasting()
674 int longer_index = longer->dimensions_count() - 1; in ShapesAgreeUpToBroadcasting()
675 int shorter_index = shorter->dimensions_count() - 1; in ShapesAgreeUpToBroadcasting()
[all …]
Dtooling_util.h144 return ::tflite::RuntimeShape(shape.dimensions_count(), shape.dims().data());
222 DCHECK_EQ(shape.dimensions_count(), indices.size());
223 const int dims_count = shape.dimensions_count();
237 const int dims_count = shape.dimensions_count();
Ddump_graphviz.cc281 for (int dim = 0; dim < array_shape.dimensions_count(); dim++) { in GetArrayLabel()
283 if (dim + 1 < array_shape.dimensions_count()) { in GetArrayLabel()
297 if (array.shape().dimensions_count() > 0) { in GetArrayLabel()
306 if (array.shape().dimensions_count() > 0) { in GetArrayLabel()
Dexport_tensorflow.cc108 const int kDims = input_shape.dimensions_count(); in ExportFloatArray()
126 CHECK_EQ(input_shape.dimensions_count(), AxesCount(input_axes_order)); in ExportFloatArray()
236 for (int i = 0; i < array_shape.dimensions_count(); i++) { in ConvertBoolTensorConst()
260 for (int i = 0; i < array_shape.dimensions_count(); i++) { in ConvertIntTensorConst()
309 for (int i = 0; i < array_shape.dimensions_count(); i++) { in ConvertComplex64TensorConst()
476 CHECK_EQ(src_weights_shape.dimensions_count(), 4); in ConvertDepthwiseConvOperator()
605 CHECK_EQ(fc_weights_shape.dimensions_count(), 2); in ConvertFullyConnectedOperator()
831 for (int i = 0; i < input_shape.dimensions_count() - 1; ++i) { in ConvertSoftmaxOperator()
835 flattened_size, input_shape.dims(input_shape.dimensions_count() - 1)}; in ConvertSoftmaxOperator()
872 for (int i = 0; i < input_shape.dimensions_count() - 1; ++i) { in ConvertLogSoftmaxOperator()
[all …]

12