Home
last modified time | relevance | path

Searched defs:input_dims (Results 1 – 25 of 130) sorted by relevance

123456

/external/tensorflow/tensorflow/lite/micro/kernels/
Dresize_nearest_neighbor_test.cc49 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestResizeNearestNeighbor() local
96 const int input_dims[] = {4, 1, 1, 2, 1}; in TF_LITE_MICRO_TEST() local
108 const int input_dims[] = {4, 1, 1, 2, 1}; in TF_LITE_MICRO_TEST() local
120 const int input_dims[] = {4, 1, 1, 2, 1}; in TF_LITE_MICRO_TEST() local
132 const int input_dims[] = {4, 1, 2, 1, 1}; in TF_LITE_MICRO_TEST() local
144 const int input_dims[] = {4, 1, 2, 1, 1}; in TF_LITE_MICRO_TEST() local
156 const int input_dims[] = {4, 1, 2, 1, 1}; in TF_LITE_MICRO_TEST() local
168 const int input_dims[] = {4, 1, 2, 2, 1}; in TF_LITE_MICRO_TEST() local
188 const int input_dims[] = {4, 1, 2, 2, 1}; in TF_LITE_MICRO_TEST() local
207 const int input_dims[] = {4, 1, 2, 2, 1}; in TF_LITE_MICRO_TEST() local
[all …]
Dl2norm_test.cc99 const int input_dims[] = {4, 1, 1, 1, 6}; in TF_LITE_MICRO_TEST() local
111 const int input_dims[] = {4, 1, 1, 1, 6}; in TF_LITE_MICRO_TEST() local
122 const int input_dims[] = {4, 1, 1, 1, 6}; in TF_LITE_MICRO_TEST() local
134 const int input_dims[] = {4, 3, 1, 1, 6}; in TF_LITE_MICRO_TEST() local
153 const int input_dims[] = {4, 1, 1, 1, 6}; in TF_LITE_MICRO_TEST() local
164 const int input_dims[] = {4, 1, 1, 1, 6}; in TF_LITE_MICRO_TEST() local
177 const int input_dims[] = {4, 1, 1, 1, 6}; in TF_LITE_MICRO_TEST() local
188 const int input_dims[] = {4, 1, 1, 1, 6}; in TF_LITE_MICRO_TEST() local
199 const int input_dims[] = {2, 3, 6}; in TF_LITE_MICRO_TEST() local
218 const int input_dims[] = {2, 3, 6}; in TF_LITE_MICRO_TEST() local
Darg_min_max_test.cc54 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestArgMinMaxFloat() local
80 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestArgMinMaxQuantized() local
107 const int input_dims[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
121 const int input_dims[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
136 const int input_dims[] = {4, 1, 1, 1, input_size}; in TF_LITE_MICRO_TEST() local
155 const int input_dims[] = {4, 1, 1, 1, input_size}; in TF_LITE_MICRO_TEST() local
174 const int input_dims[] = {4, 1, 1, 1, input_size}; in TF_LITE_MICRO_TEST() local
193 const int input_dims[] = {4, 1, 1, 1, input_size}; in TF_LITE_MICRO_TEST() local
212 const int input_dims[] = {4, 1, 1, 2, 4}; in TF_LITE_MICRO_TEST() local
231 const int input_dims[] = {4, 1, 1, 2, 4}; in TF_LITE_MICRO_TEST() local
[all …]
Dzeros_like_test.cc29 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestZerosLikeFloat() local
61 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestZerosLikeInt32() local
93 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestZerosLikeInt64() local
130 const int input_dims[] = {2, 2, 3}; in TF_LITE_MICRO_TEST() local
139 const int input_dims[] = {4, 1, 2, 2, 1}; in TF_LITE_MICRO_TEST() local
148 const int input_dims[] = {4, 1, 2, 2, 1}; in TF_LITE_MICRO_TEST() local
Dcast_test.cc30 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestCastFloatToInt8() local
62 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestCastInt8ToFloat() local
99 const int input_dims[] = {2, 3, 2}; in TF_LITE_MICRO_TEST() local
110 const int input_dims[] = {2, 3, 2}; in TF_LITE_MICRO_TEST() local
Dpad_test.cc97 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestPadFloat() local
122 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestPadV2Float() local
152 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestPadQuantized() local
188 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestPadV2Quantized() local
227 const int input_dims[] = {4, 1, 2, 2, 1}; in TF_LITE_MICRO_TEST() local
241 const int input_dims[] = {4, 1, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
260 const int input_dims[] = {4, 1, 2, 2, 1}; in TF_LITE_MICRO_TEST() local
276 const int input_dims[] = {4, 1, 2, 2, 1}; in TF_LITE_MICRO_TEST() local
298 const int input_dims[] = {4, 1, 2, 2, 1}; in TF_LITE_MICRO_TEST() local
324 const int input_dims[] = {4, 1, 2, 2, 1}; in TF_LITE_MICRO_TEST() local
[all …]
Dround_test.cc29 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestRound() local
64 const int input_dims[] = {1, 6}; in TF_LITE_MICRO_TEST() local
72 const int input_dims[] = {4, 2, 1, 1, 6}; in TF_LITE_MICRO_TEST() local
Dceil_test.cc29 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestCeil() local
66 const int input_dims[] = {1, 2}; in TF_LITE_MICRO_TEST() local
74 const int input_dims[] = {4, 2, 1, 1, 5}; in TF_LITE_MICRO_TEST() local
Dreshape_test.cc121 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestReshape() local
142 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestReshapeQuantized() local
163 const int input_dims[] = {4, 1, 2, 4, 1}; in TF_LITE_MICRO_TEST() local
179 const int input_dims[] = {9, 1, 1, 2, 1, 1, 1, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
195 const int input_dims[] = {4, 1, 2, 4, 11}; in TF_LITE_MICRO_TEST() local
213 TfLiteIntArray* input_dims = in TF_LITE_MICRO_TEST() local
235 const int input_dims[] = {4, 1, 2, 4, 1}; in TF_LITE_MICRO_TEST() local
267 const int input_dims[] = {4, 1, 2, 4, 1}; in TF_LITE_MICRO_TEST() local
299 const int input_dims[] = {1, 1}; in TF_LITE_MICRO_TEST() local
333 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TF_LITE_MICRO_TEST() local
Dexp_test.cc30 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestExp() local
66 const int input_dims[] = {2, 1, kInputSize}; in TF_LITE_MICRO_TEST() local
Dactivations_test.cc30 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestReluFloat() local
63 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestRelu6Float() local
99 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestReluUint8() local
140 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestRelu6Uint8() local
181 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestReluInt8() local
221 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestRelu6Int8() local
Dcircular_buffer_test.cc48 const int input_dims[] = {4, 1, 1, 1, depth}; in TF_LITE_MICRO_TEST() local
118 const int input_dims[] = {4, 1, 1, 1, depth}; in TF_LITE_MICRO_TEST() local
Dtranspose_conv_test.cc105 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestTransposeConvFloat() local
141 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestTransposeConvQuantized() local
229 TfLiteIntArray* input_dims = IntArrayFromInts(tflite::testing::kInputShape); in TF_LITE_MICRO_TEST() local
263 TfLiteIntArray* input_dims = IntArrayFromInts(tflite::testing::kInputShape); in TF_LITE_MICRO_TEST() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h42 inline void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
69 inline void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
85 void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
101 void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
111 inline void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
146 inline void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
166 void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
189 void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
206 inline void Conv(const float* input_data, const Dims<4>& input_dims, in Conv()
232 void Conv(const float* input_data, const Dims<4>& input_dims, in Conv()
[all …]
Dreduce.h34 inline bool Reduce(const In* input_data, const int* input_dims, in Reduce()
93 inline bool ReduceSumImpl(const In* input_data, const int* input_dims, in ReduceSumImpl()
128 inline bool ReduceGeneric(const T* input_data, const int* input_dims, in ReduceGeneric()
162 inline bool Mean(const T* input_data, const int* input_dims, in Mean()
324 float input_scale, const int* input_dims, in QuantizedMeanOrSum()
/external/tensorflow/tensorflow/core/kernels/
Deigen_benchmark.h46 void SpatialConvolution(Dimensions input_dims, Dimensions filter_dims) { in SpatialConvolution()
76 void SpatialConvolutionBackwardInput(Dimensions input_dims, in SpatialConvolutionBackwardInput()
115 void SpatialConvolutionBackwardKernel(Dimensions input_dims, in SpatialConvolutionBackwardKernel()
177 void CuboidConvolution(Dimensions input_dims, Dimensions filter_dims) { in CuboidConvolution()
208 void CuboidConvolutionBackwardInput(Dimensions input_dims, in CuboidConvolutionBackwardInput()
249 void CuboidConvolutionBackwardKernel(Dimensions input_dims, in CuboidConvolutionBackwardKernel()
Dreduce_join_op.cc77 int32 input_dims) { in GetReducedIndices()
92 int32 input_dims, in MakeUnreducedIndices()
127 const int32 input_dims = input_shape.dims(); in Compute() local
Dtile_ops.cc175 const int input_dims = input.dims(); in Compute() local
341 const int input_dims = input.dims(); in Compute() local
421 const std::vector<Tmultiples>& input_dims, in HandleCaseImpl()
514 OpKernelContext* context, const std::vector<Tmultiples>& input_dims, in HandleCase()
Dwhere_op.cc141 const int input_dims = input.dims(); in Compute() local
263 const int input_dims = input.dims(); in ComputeAsync() local
273 void ComputeAsyncType(const Tensor& input, const int input_dims, in ComputeAsyncType()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dquantization_util_test.cc149 auto input_dims = BuildTfLiteIntArray({1, 3, 2, 1}); in TEST() local
174 auto input_dims = BuildTfLiteIntArray({1, 3, 2, 1}); in TEST() local
199 auto input_dims = BuildTfLiteIntArray({1, 3, 2, 1}); in TEST() local
221 auto input_dims = BuildTfLiteIntArray({1, 3, 2, 1}); in TEST() local
/external/libtextclassifier/native/utils/tflite/
Dblacklist_base.cc39 TfLiteIntArray* input_dims = op->GetInputShape(context, node); in Resize() local
56 TfLiteIntArray* input_dims = op->GetInputShape(context, node); in Eval() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h138 inline void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
170 inline void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
186 void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
202 void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
280 inline void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
320 inline void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
340 void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
363 void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
684 inline void FullyConnected(const float* input_data, const Dims<4>& input_dims, in FullyConnected()
703 void FullyConnected(const float* input_data, const Dims<4>& input_dims, in FullyConnected()
[all …]
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_test.cc245 std::vector<int64> input_dims = {1, 4, 2, 3, 3}; in XLA_TEST_F() local
311 std::vector<int64> input_dims = {1, 3, 3, 5}; in RunTest() local
370 std::vector<int64> input_dims = {1, 3, 3, 5}; in RunTest() local
436 std::vector<int64> input_dims = {1, 4, 4, 5}; in RunTest() local
506 std::vector<int64> input_dims = {1, 4, 4, 512}; in RunTest() local
570 std::vector<int64> input_dims = {1, 4, 4, 512}; in RunTest() local
639 std::vector<int64> input_dims = {256, 4, 4, 512}; in RunTest() local
708 std::vector<int64> input_dims = {256, 4, 4, 512}; in RunTest() local
779 std::vector<int64> input_dims = {1, 4, 4, 5}; in RunTest() local
854 std::vector<int64> input_dims = {1, 4, 4, 160}; in RunTest() local
[all …]
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dreorder_reshape_transpose.cc58 std::vector<int> ComputeNewPerm(std::vector<int> input_dims, in ComputeNewPerm()
171 std::vector<int> input_dims = input_shape.dims(); in Run() local
/external/tensorflow/tensorflow/lite/kernels/
Dmirror_pad.cc48 const TfLiteIntArray* input_dims = nullptr; member
85 const int input_dims = NumDimensions(input); in GetPaddedOutputShape() local
175 const int input_dims = NumDimensions(input_tensor); in Eval() local

123456