Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Ddensify.h33 const int dims_count = output_shape.DimensionsCount(); in Densify() local
Dprocess_broadcast_shapes.h41 const int dims_count = in ProcessBroadcastShapes() local
Darg_min_max.h44 const int dims_count = input1_shape.DimensionsCount(); in ArgMinMax() local
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dbinary_elementwise_tester.cc266 int dims_count = Input1Shape().size(); in CreateTfLiteModel() local
292 int dims_count = Input2Shape().size(); in CreateTfLiteModel() local
Dprelu_tester.cc148 const int dims_count = SlopeShape().size(); in CreateTfLiteModel() local
Dconv_2d_tester.cc256 int dims_count = filter_shape.size(); in CreateTfLiteModel() local
Ddepthwise_conv_2d_tester.cc258 int dims_count = filter_shape.size(); in CreateTfLiteModel() local
Dxnnpack_delegate.cc3173 const int dims_count = output_tensor.dims->size; in PrepareOpsToDelegate() local
/external/tensorflow/tensorflow/lite/micro/kernels/
Dprelu_test.cc150 const int dims_count = 12; in TF_LITE_MICRO_TEST() local
182 const int dims_count = 12; in TF_LITE_MICRO_TEST() local
/external/tensorflow/tensorflow/lite/kernels/internal/
Dstrided_slice_logic.h187 const int dims_count = start_indices.size(); in BuildStridedSliceParams() local
Dtypes.h522 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local
533 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local
545 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local
558 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local
674 const int dims_count = shape.DimensionsCount(); in FlatSizeSkipDim() local
687 const int dims_count = shape.DimensionsCount(); in MatchingFlatSizeSkipDim() local
699 const int dims_count = shape.DimensionsCount(); in MatchingFlatSizeSkipDim() local
712 const int dims_count = shape.DimensionsCount(); in MatchingFlatSizeSkipDim() local
726 const int dims_count = shape.DimensionsCount(); in MatchingFlatSizeSkipDim() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Ddense_to_sparse.cc54 const int dims_count = block_size.size(); in PopulateEncodingParams() local
231 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.cc76 int dims_count = in GetDeviceWorkDimsSizes() local
/external/tensorflow/tensorflow/lite/kernels/
Dtest_util.h221 const int dims_count = t.traversal_order.size(); in AddConstSparseInput() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h150 const int dims_count = shape.DimensionsCount(); in MapAsMatrixWithLastDimAsRows() local
174 const int dims_count = shape.DimensionsCount(); in MapAsArrayWithLastDimAsRows() local
339 const int dims_count = weights_shape.DimensionsCount(); in FullyConnected() local
Dlegacy_optimized_ops.h660 const int dims_count = weights_shape.DimensionsCount(); in FullyConnected() local