/external/tensorflow/tensorflow/core/kernels/ |
D | eigen_benchmark.h | 46 void SpatialConvolution(Dimensions input_dims, Dimensions filter_dims) { in SpatialConvolution() argument 47 Dimensions output_dims(input_dims[0], // batch in SpatialConvolution() 48 input_dims[1], // input_height in SpatialConvolution() 49 input_dims[2], // input_width in SpatialConvolution() 53 static_cast<Scalar*>(device_.allocate(BufferSize(input_dims))); in SpatialConvolution() 59 device_.memset(input_data, 123, BufferSize(input_dims)); in SpatialConvolution() 62 Input input(input_data, input_dims); in SpatialConvolution() 76 void SpatialConvolutionBackwardInput(Dimensions input_dims, in SpatialConvolutionBackwardInput() argument 81 Dimensions output_dims(input_dims[0], // batch in SpatialConvolutionBackwardInput() 82 input_dims[1], // input_height in SpatialConvolutionBackwardInput() [all …]
|
D | conv_2d_gpu.h | 187 Dimension<3> input_dims, 190 output_dims[sp0] = input_dims[0]; 191 output_dims[sp1] = input_dims[1]; 192 output_dims[sp2] = input_dims[2]; 206 int input_index = TensorIndexToFlat(input_tensor_index, input_dims); 218 Dimension<3> input_dims, 221 output_dims[sp0] = input_dims[0]; 222 output_dims[sp1] = input_dims[1]; 223 output_dims[sp2] = input_dims[2]; 242 int input_index_i = TensorIndexToFlat(input_tensor_index, input_dims); [all …]
|
D | reduce_join_op.cc | 77 int32 input_dims) { in GetReducedIndices() argument 84 reduced_indices[i] += reduced_indices[i] < 0 ? input_dims : 0; in GetReducedIndices() 92 int32 input_dims, in MakeUnreducedIndices() argument 94 for (int32 index = 0; index < input_dims; ++index) { in MakeUnreducedIndices() 127 const int32 input_dims = input_shape.dims(); in Compute() local 133 gtl::InlinedVector<bool, 8> index_is_reduced(input_dims, false); in Compute() 137 reduce_index < 0 ? reduce_index + input_dims : reduce_index; in Compute() 139 context, reduce_index >= -input_dims && reduce_index < input_dims, in Compute() 141 " for input with ", input_dims, " dimension(s)")); in Compute() 149 GetReducedIndices(reduction_indices, input_dims); in Compute() [all …]
|
D | sparse_slice_op.cc | 58 const int input_dims = input_shape.NumElements(); in Compute() local 59 OP_REQUIRES(context, input_dims == input_start.NumElements(), in Compute() 61 "Expected start to be a vector of length ", input_dims, in Compute() 64 OP_REQUIRES(context, input_dims == input_size.NumElements(), in Compute() 66 "Expected size to be a vector of length ", input_dims, in Compute() 76 input_dims); in Compute() 78 input_dims); in Compute()
|
/external/tensorflow/tensorflow/lite/micro/kernels/ |
D | resize_nearest_neighbor_test.cc | 49 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestResizeNearestNeighbor() local 61 TestCreateTensor(input_data, input_dims), in TestResizeNearestNeighbor() 96 const int input_dims[] = {4, 1, 1, 2, 1}; in TF_LITE_MICRO_TEST() local 104 input_dims, input_data, expected_size_data, expected_output_data, in TF_LITE_MICRO_TEST() 108 const int input_dims[] = {4, 1, 1, 2, 1}; in TF_LITE_MICRO_TEST() local 116 input_dims, input_data, expected_size_data, expected_output_data, in TF_LITE_MICRO_TEST() 120 const int input_dims[] = {4, 1, 1, 2, 1}; in TF_LITE_MICRO_TEST() local 128 input_dims, input_data, expected_size_data, expected_output_data, in TF_LITE_MICRO_TEST() 132 const int input_dims[] = {4, 1, 2, 1, 1}; in TF_LITE_MICRO_TEST() local 140 input_dims, input_data, expected_size_data, expected_output_data, in TF_LITE_MICRO_TEST() [all …]
|
D | l2norm_test.cc | 99 const int input_dims[] = {4, 1, 1, 1, 6}; in TF_LITE_MICRO_TEST() local 107 input_dims, input_data, expected_output_data, output_data); in TF_LITE_MICRO_TEST() 111 const int input_dims[] = {4, 1, 1, 1, 6}; in TF_LITE_MICRO_TEST() local 118 input_dims, input_data, expected_output_data, output_data); in TF_LITE_MICRO_TEST() 122 const int input_dims[] = {4, 1, 1, 1, 6}; in TF_LITE_MICRO_TEST() local 130 input_dims, input_data, expected_output_data, output_data); in TF_LITE_MICRO_TEST() 134 const int input_dims[] = {4, 3, 1, 1, 6}; in TF_LITE_MICRO_TEST() local 149 input_dims, input_data, expected_output_data, output_data); in TF_LITE_MICRO_TEST() 153 const int input_dims[] = {4, 1, 1, 1, 6}; in TF_LITE_MICRO_TEST() local 159 tflite::testing::TestL2Normalization<uint8_t>(input_dims, input_data, in TF_LITE_MICRO_TEST() [all …]
|
D | arg_min_max_test.cc | 54 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestArgMinMaxFloat() local 63 CreateTensor(input_values, input_dims), in TestArgMinMaxFloat() 80 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestArgMinMaxQuantized() local 89 CreateQuantizedTensor(input_values, input_quantized, input_dims, in TestArgMinMaxQuantized() 107 const int input_dims[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local 114 tflite::testing::TestArgMinMaxFloat(input_dims, input_values, axis_dims, in TF_LITE_MICRO_TEST() 121 const int input_dims[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local 128 tflite::testing::TestArgMinMaxFloat(input_dims, input_values, axis_dims, in TF_LITE_MICRO_TEST() 136 const int input_dims[] = {4, 1, 1, 1, input_size}; in TF_LITE_MICRO_TEST() local 148 input_dims, input_values, input_quantized, input_scale, input_zero_point, in TF_LITE_MICRO_TEST() [all …]
|
D | reshape_test.cc | 121 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestReshape() local 124 TfLiteTensor input_tensor = CreateTensor(input_data, input_dims); in TestReshape() 142 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestReshapeQuantized() local 146 input_data, input_dims, /*scale=*/1.f, /*zero_point=*/0); in TestReshapeQuantized() 163 const int input_dims[] = {4, 1, 2, 4, 1}; in TF_LITE_MICRO_TEST() local 173 input_dims, input_data, shape_dims, shape_int32, output_dims, output_data, in TF_LITE_MICRO_TEST() 179 const int input_dims[] = {9, 1, 1, 2, 1, 1, 1, 1, 1, 1}; in TF_LITE_MICRO_TEST() local 189 input_dims, input, shape_dims, shape_int32, output_dims, output_data, in TF_LITE_MICRO_TEST() 195 const int input_dims[] = {4, 1, 2, 4, 11}; in TF_LITE_MICRO_TEST() local 205 input_dims, input, shape_dims, shape_int32, output_dims, output_data, in TF_LITE_MICRO_TEST() [all …]
|
D | pad_test.cc | 97 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestPadFloat() local 104 TfLiteTensor tensors[tensors_size] = {CreateTensor(input_data, input_dims), in TestPadFloat() 122 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestPadV2Float() local 132 CreateTensor(input_data, input_dims), CreateTensor(pad_data, pad_dims), in TestPadV2Float() 152 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestPadQuantized() local 160 CreateQuantizedTensor(input_data, input_quantized, input_dims, in TestPadQuantized() 188 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestPadV2Quantized() local 199 CreateQuantizedTensor(input_data, input_quantized, input_dims, in TestPadV2Quantized() 227 const int input_dims[] = {4, 1, 2, 2, 1}; in TF_LITE_MICRO_TEST() local 236 tflite::testing::TestPadFloat(input_dims, input_values, pad_dims, pad_values, in TF_LITE_MICRO_TEST() [all …]
|
D | zeros_like_test.cc | 29 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestZerosLikeFloat() local 36 CreateTensor(input_data, input_dims), in TestZerosLikeFloat() 61 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestZerosLikeInt32() local 68 CreateTensor(input_data, input_dims), in TestZerosLikeInt32() 93 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestZerosLikeInt64() local 100 CreateTensor(input_data, input_dims), in TestZerosLikeInt64() 130 const int input_dims[] = {2, 2, 3}; in TF_LITE_MICRO_TEST() local 133 tflite::testing::TestZerosLikeFloat(input_dims, input_values, golden, in TF_LITE_MICRO_TEST() 139 const int input_dims[] = {4, 1, 2, 2, 1}; in TF_LITE_MICRO_TEST() local 142 tflite::testing::TestZerosLikeInt32(input_dims, input_values, golden, in TF_LITE_MICRO_TEST() [all …]
|
D | cast_test.cc | 30 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestCastFloatToInt8() local 37 CreateTensor(input_data, input_dims), in TestCastFloatToInt8() 62 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestCastInt8ToFloat() local 69 CreateTensor(input_data, input_dims), in TestCastInt8ToFloat() 99 const int input_dims[] = {2, 3, 2}; in TF_LITE_MICRO_TEST() local 104 tflite::testing::TestCastFloatToInt8(input_dims, input_values, golden, in TF_LITE_MICRO_TEST() 110 const int input_dims[] = {2, 3, 2}; in TF_LITE_MICRO_TEST() local 113 tflite::testing::TestCastInt8ToFloat(input_dims, input_values, golden, in TF_LITE_MICRO_TEST()
|
D | round_test.cc | 29 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestRound() local 36 CreateTensor(input_data, input_dims), in TestRound() 64 const int input_dims[] = {1, 6}; in TF_LITE_MICRO_TEST() local 68 tflite::testing::TestRound(input_dims, input_data, golden, output_data); in TF_LITE_MICRO_TEST() 72 const int input_dims[] = {4, 2, 1, 1, 6}; in TF_LITE_MICRO_TEST() local 77 tflite::testing::TestRound(input_dims, input_data, golden, output_data); in TF_LITE_MICRO_TEST()
|
D | ceil_test.cc | 29 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestCeil() local 36 CreateTensor(input_data, input_dims), in TestCeil() 66 const int input_dims[] = {1, 2}; in TF_LITE_MICRO_TEST() local 69 tflite::testing::TestCeil(input_dims, input_values, golden, output_data); in TF_LITE_MICRO_TEST() 74 const int input_dims[] = {4, 2, 1, 1, 5}; in TF_LITE_MICRO_TEST() local 80 tflite::testing::TestCeil(input_dims, input_values, golden, output_data); in TF_LITE_MICRO_TEST()
|
/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
D | legacy_reference_ops.h | 42 inline void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv() argument 64 DepthwiseConv(op_params, DimsToShape(input_dims), input_data, in DepthwiseConv() 69 inline void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv() argument 77 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv() 85 void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv() argument 93 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv() 101 void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv() argument 106 DepthwiseConv<Ac>(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv() 111 inline void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv() argument 141 DepthwiseConv(op_params, DimsToShape(input_dims), input_data, in DepthwiseConv() [all …]
|
D | reduce.h | 34 inline bool Reduce(const In* input_data, const int* input_dims, in Reduce() argument 47 ReducedOutputOffset(input_num_dims, input_dims, input_iter, 0, nullptr); in Reduce() 48 size_t output_offset = ReducedOutputOffset(input_num_dims, input_dims, in Reduce() 52 } while (NextIndex(input_num_dims, input_dims, input_iter)); in Reduce() 93 inline bool ReduceSumImpl(const In* input_data, const int* input_dims, in ReduceSumImpl() argument 102 return Reduce<In, Out>(input_data, input_dims, output_dims, input_num_dims, in ReduceSumImpl() 128 inline bool ReduceGeneric(const T* input_data, const int* input_dims, in ReduceGeneric() argument 137 if (input_dims[i] == 0) return true; in ReduceGeneric() 153 return Reduce<T, T>(input_data, input_dims, output_dims, input_num_dims, in ReduceGeneric() 162 inline bool Mean(const T* input_data, const int* input_dims, in Mean() argument [all …]
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | slice_op.cc | 46 const int input_dims = input_shape.dims(); in Compile() local 51 begin_tensor_shape.num_elements() == input_dims && in Compile() 52 size_tensor_shape.num_elements() == input_dims, in Compile() 55 input_dims, ", but got shapes ", begin_tensor_shape.DebugString(), in Compile() 66 for (int i = 0; i < input_dims; ++i) { in Compile() 75 for (int i = 0; i < input_dims; ++i) { in Compile() 124 for (int i = 0; i < input_dims; ++i) { in Compile() 138 scalar_indices.reserve(input_dims); in Compile() 140 for (int i = 0; i < input_dims; i++) { in Compile() 154 for (xla::int64 i = 0; i < input_dims; ++i) { in Compile() [all …]
|
D | roll_op.cc | 34 int64 input_dims = input_shape.dims(); in Compile() local 35 OP_REQUIRES(ctx, input_dims >= 1, in Compile() 58 OP_REQUIRES(ctx, cur_axis >= -input_dims && cur_axis < input_dims, in Compile() 61 input_dims, ", ", input_dims, ")."))); in Compile() 63 cur_axis += input_dims; in Compile()
|
/external/libtextclassifier/native/utils/tflite/ |
D | blacklist_base.cc | 39 TfLiteIntArray* input_dims = op->GetInputShape(context, node); in Resize() local 40 TfLiteIntArray* output_dims = TfLiteIntArrayCreate(input_dims->size + 1); in Resize() 41 for (int i = 0; i < input_dims->size; i++) { in Resize() 42 output_dims->data[i] = input_dims->data[i]; in Resize() 44 output_dims->data[input_dims->size] = op->categories(); in Resize() 56 TfLiteIntArray* input_dims = op->GetInputShape(context, node); in Eval() local 58 for (int i = 0; i < input_dims->size; i++) { in Eval() 59 input_size *= input_dims->data[i]; in Eval()
|
/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
D | quantization_util_test.cc | 149 auto input_dims = BuildTfLiteIntArray({1, 3, 2, 1}); in TEST() local 154 data.data(), input_dims.get(), "input", in TEST() 157 dequantized_data.data(), input_dims.get(), "input_dequant", in TEST() 174 auto input_dims = BuildTfLiteIntArray({1, 3, 2, 1}); in TEST() local 179 CreateQuantizedTensor(data.data(), input_dims.get(), "input", in TEST() 182 dequantized_data.data(), input_dims.get(), "input_dequant", in TEST() 199 auto input_dims = BuildTfLiteIntArray({1, 3, 2, 1}); in TEST() local 202 TfLiteTensor output = CreateFloatTensor(data.data(), input_dims.get(), in TEST() 205 quantized_data.data(), input_dims.get(), "output_quant", in TEST() 221 auto input_dims = BuildTfLiteIntArray({1, 3, 2, 1}); in TEST() local [all …]
|
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/ |
D | conv.cc | 63 cmsis_nn_dims input_dims; in Prepare() local 64 input_dims.n = MatchingDim(input_shape, 0, output_shape, 0); in Prepare() 65 input_dims.h = input->dims->data[1]; in Prepare() 66 input_dims.w = input->dims->data[2]; in Prepare() 67 input_dims.c = input_shape.Dims(3); in Prepare() 74 filter_dims.c = input_dims.c; in Prepare() 78 output_dims.n = input_dims.n; in Prepare() 97 context, node, params, input_dims.w, input_dims.h, filter_dims.w, in Prepare() 116 &conv_params, &input_dims, &filter_dims, &output_dims); in Prepare() 176 cmsis_nn_dims input_dims; in EvalQuantizedPerChannel() local [all …]
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | mirror_pad.cc | 48 const TfLiteIntArray* input_dims = nullptr; member 85 const int input_dims = NumDimensions(input); in GetPaddedOutputShape() local 87 TfLiteIntArrayCreate(input_dims), TfLiteIntArrayFree); in GetPaddedOutputShape() 90 for (int i = 0; i < input_dims; ++i) { in GetPaddedOutputShape() 136 eval_data->input_dims->data[i], eval_data->offset); in GetFlatIndex() 175 const int input_dims = NumDimensions(input_tensor); in Eval() local 188 std::vector<int> output_dims_num_elements(input_dims, 1); in Eval() 189 std::vector<int> input_dims_num_elements(input_dims, 1); in Eval() 190 for (int i = input_dims - 2; i >= 0; i--) { in Eval() 209 eval_data.input_dims = input_tensor->dims; \ in Eval() [all …]
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | convolution_test.cc | 245 std::vector<int64> input_dims = {1, 4, 2, 3, 3}; in XLA_TEST_F() local 247 Shape input_shape = ShapeUtil::MakeShape(F32, input_dims); in XLA_TEST_F() 277 auto input_r5 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in XLA_TEST_F() 311 std::vector<int64> input_dims = {1, 3, 3, 5}; in RunTest() local 313 Shape input_shape = ShapeUtil::MakeShapeWithType<T>(input_dims); in RunTest() 340 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in RunTest() 370 std::vector<int64> input_dims = {1, 3, 3, 5}; in RunTest() local 372 Shape input_shape = ShapeUtil::MakeShapeWithType<T>(input_dims); in RunTest() 400 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in RunTest() 436 std::vector<int64> input_dims = {1, 4, 4, 5}; in RunTest() local [all …]
|
D | reverse_test.cc | 43 std::vector<int64> input_dims; member 49 "reverse_%s_in_dims_%s_%s", absl::StrJoin(input_dims, "x"), in ToTestCaseName() 84 ShapeUtil::ElementsIn(ShapeUtil::MakeShape(F32, spec.input_dims))); in TEST_P() 87 auto input_literal = r1_literal.Reshape(spec.input_dims).ConsumeValueOrDie(); in TEST_P() 94 std::vector<int64> output_indices(spec.input_dims.size()); in TEST_P() 101 output_indices[dim] = (spec.input_dims[dim] - 1) - indices[dim]; in TEST_P()
|
/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
D | reorder_reshape_transpose.cc | 58 std::vector<int> ComputeNewPerm(std::vector<int> input_dims, in ComputeNewPerm() argument 63 for (size_t i = 0; i < input_dims.size(); i++) { in ComputeNewPerm() 64 if (input_dims[i] != 1) { in ComputeNewPerm() 82 new_perm.reserve(input_dims.size()); in ComputeNewPerm() 90 for (size_t index = 0; index < input_dims.size(); index++) { in ComputeNewPerm() 91 if (input_dims[index] == 1) { in ComputeNewPerm() 96 CHECK_EQ(new_perm.size(), input_dims.size()); in ComputeNewPerm() 171 std::vector<int> input_dims = input_shape.dims(); in Run() local 217 ComputeNewPerm(input_dims, intermediate_dims, transpose_op->perm); in Run() 218 CHECK_EQ(input_dims.size(), new_perm.size()); in Run()
|
/external/XNNPACK/test/ |
D | constant-pad-operator-tester.h | 114 std::array<size_t, XNN_MAX_TENSOR_DIMS> input_dims; in TestX32() local 118 std::fill(input_dims.begin(), input_dims.end(), 1); in TestX32() 123 input_dims[XNN_MAX_TENSOR_DIMS - num_dims() + i] = input_dim(i); in TestX32() 136 input_stride *= input_dims[i - 1]; in TestX32() 150 for (size_t i = 0; i < input_dims[0]; i++) { in TestX32() 151 for (size_t j = 0; j < input_dims[1]; j++) { in TestX32() 152 for (size_t k = 0; k < input_dims[2]; k++) { in TestX32() 153 for (size_t l = 0; l < input_dims[3]; l++) { in TestX32() 154 for (size_t m = 0; m < input_dims[4]; m++) { in TestX32() 155 for (size_t n = 0; n < input_dims[5]; n++) { in TestX32()
|