Home
last modified time | relevance | path

Searched defs:dims_count (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops_utils.h58 const int dims_count = shape.DimensionsCount(); in MapAsMatrixWithLastDimAsRows() local
82 const int dims_count = shape.DimensionsCount(); in MapAsArrayWithLastDimAsRows() local
Doptimized_ops.h275 const int dims_count = weights_shape.DimensionsCount(); in FullyConnected() local
8003 const int dims_count = input1_shape.DimensionsCount(); in ArgMinMax() local
Dlegacy_optimized_ops.h661 const int dims_count = weights_shape.DimensionsCount(); in FullyConnected() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Ddensify.h33 const int dims_count = output_shape.DimensionsCount(); in Densify() local
Dprocess_broadcast_shapes.h43 const int dims_count = in ProcessBroadcastShapes() local
Darg_min_max.h55 const int dims_count = input1_shape.DimensionsCount(); in ArgMinMax() local
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dbinary_elementwise_tester.cc356 int dims_count = Input1Shape().size(); in CreateTfLiteModel() local
393 int dims_count = Input2Shape().size(); in CreateTfLiteModel() local
Dprelu_tester.cc201 const int dims_count = SlopeShape().size(); in CreateTfLiteModel() local
Dconv_2d_tester.cc308 int dims_count = filter_shape.size(); in CreateTfLiteModel() local
Ddepthwise_conv_2d_tester.cc317 int dims_count = filter_shape.size(); in CreateTfLiteModel() local
Dtranspose_conv_tester.cc305 const int dims_count = filter_shape.size(); in CreateTfLiteModel() local
Dxnnpack_delegate.cc4411 const int dims_count = NumDimensions(&input_tensor); in VisitTransposeNode() local
5035 const int dims_count = NumDimensions(&output_tensor); in PrepareOpsToDelegate() local
/external/tensorflow/tensorflow/lite/kernels/internal/
Dstrided_slice_logic.h187 const int dims_count = start_indices.size(); in BuildStridedSliceParams() local
Dtypes.h315 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local
326 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local
338 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local
351 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local
519 const int dims_count = shape.DimensionsCount(); in FlatSizeSkipDim() local
532 const int dims_count = shape.DimensionsCount(); in MatchingFlatSizeSkipDim() local
544 const int dims_count = shape.DimensionsCount(); in MatchingFlatSizeSkipDim() local
557 const int dims_count = shape.DimensionsCount(); in MatchingFlatSizeSkipDim() local
571 const int dims_count = shape.DimensionsCount(); in MatchingFlatSizeSkipDim() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Ddense_to_sparse.cc62 const int dims_count = block_size.size(); in PopulateEncodingParams() local
239 const int dims_count = type.getRank(); in BuildSparsityParameterAttribute() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_binary.cc95 const int dims_count = output_shape.dimensions_count(); in EvaluateBinaryOperatorOnConstantInputs() local
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dcl_device.cc109 int dims_count = in GetDeviceWorkDimsSizes() local
/external/tensorflow/tensorflow/lite/kernels/
Dtest_util.h253 const int dims_count = t.traversal_order.size(); in AddConstSparseInput() local
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc5286 const int dims_count = output_tensor.dims->size; in DensifyAndDequantizeConstTensor() local