Home
last modified time | relevance | path

Searched refs:output_array (Results 1 – 25 of 95) sorted by relevance

1234

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dresolve_constant_reshape.cc45 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
46 if (output_array.data_type == ArrayDataType::kNone) { in Run()
50 if (!output_array.has_shape()) { in Run()
56 if (!ShapesAgreeUpToExtending(input_array.shape(), output_array.shape())) { in Run()
59 ShapeToString(output_array.shape())); in Run()
63 CHECK(!output_array.buffer); in Run()
66 CopyArrayBuffer<ArrayDataType::kBool>(input_array, &output_array); in Run()
69 CopyArrayBuffer<ArrayDataType::kFloat>(input_array, &output_array); in Run()
72 CopyArrayBuffer<ArrayDataType::kInt8>(input_array, &output_array); in Run()
75 CopyArrayBuffer<ArrayDataType::kUint8>(input_array, &output_array); in Run()
[all …]
H A Dpropagate_fixed_sizes.cc86 Array* output_array) { in ComputeBinaryOperatorOutputSize() argument
94 std::vector<int>* dims_out = output_array->mutable_shape()->mutable_dims(); in ComputeBinaryOperatorOutputSize()
122 CHECK(output_array->has_shape()); in ComputeBinaryOperatorOutputSize()
144 auto& output_array = model->GetArray(op->outputs[0]); in ProcessConvOperator() local
151 output_array.mutable_shape(), in ProcessConvOperator()
153 CHECK_EQ(output_array.shape().dimensions_count(), 4); in ProcessConvOperator()
157 const auto& output_shape = output_array.shape(); in ProcessConvOperator()
239 auto& output_array = model->GetArray(op->outputs[0]); in ProcessTransposeConvOperator() local
240 *(output_array.mutable_shape()->mutable_dims()) = specified_output_shape; in ProcessTransposeConvOperator()
343 auto& output_array = model->GetArray(op->outputs[0]); in ProcessOpWithShapeInput() local
[all …]
H A Dresolve_constant_tile.cc74 Array* output_array) { in Tile() argument
76 auto& output_data = output_array->GetMutableBuffer<Type>().data; in Tile()
77 output_data.resize(RequiredBufferSizeForShape(output_array->shape())); in Tile()
84 output_array->GetMutableBuffer<Type>().data.data(), 0); in Tile()
90 output_array->GetMutableBuffer<Type>().data.data(), 0); in Tile()
114 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
115 if (output_array.data_type == ArrayDataType::kNone) { in Run()
119 if (!output_array.has_shape()) { in Run()
135 CopyMinMaxAndQuantizationRelatedFields(input_array, &output_array); in Run()
137 CHECK(!output_array.buffer); in Run()
[all …]
H A Dresolve_constant_gather.cc29 Array* output_array) { in Gather() argument
37 const Shape& output_shape = output_array->shape(); in Gather()
39 output_array->GetMutableBuffer<Type>().data; in Gather()
42 CHECK_EQ(coords_shape.dims(0), output_array->shape().dims(0)); in Gather()
80 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
81 if (output_array.data_type == ArrayDataType::kNone) { in Run()
85 if (!output_array.has_shape()) { in Run()
116 auto& output_minmax = output_array.GetOrCreateMinMax(); in Run()
121 CHECK(!output_array.buffer); in Run()
122 switch (output_array.data_type) { in Run()
[all …]
H A Dhardcode_min_max.cc50 auto& output_array = model->GetArray(op->outputs[0]); in HardcodeMinMaxForL2Normalization() local
51 if (output_array.minmax) { in HardcodeMinMaxForL2Normalization()
59 CHECK(!output_array.minmax); in HardcodeMinMaxForL2Normalization()
60 auto& output_minmax = output_array.GetOrCreateMinMax(); in HardcodeMinMaxForL2Normalization()
173 auto& output_array = model->GetArray(op->outputs[0]); in HardcodeMinMaxForAverageOrMaxPool() local
174 if (output_array.minmax) { in HardcodeMinMaxForAverageOrMaxPool()
182 CHECK(!output_array.minmax); in HardcodeMinMaxForAverageOrMaxPool()
183 auto& output_minmax = output_array.GetOrCreateMinMax(); in HardcodeMinMaxForAverageOrMaxPool()
190 auto& output_array = model->GetArray(op->outputs[0]); in HardcodeMinMaxFromFirstInput() local
191 if (output_array.minmax) { in HardcodeMinMaxFromFirstInput()
[all …]
H A Dresolve_constant_strided_slice.cc29 Array* output_array) { in StridedSlice() argument
36 CHECK(output_array->data_type == Type); in StridedSlice()
47 output_array->GetMutableBuffer<Type>().data; in StridedSlice()
48 output_data.resize(RequiredBufferSizeForShape(output_array->shape())); in StridedSlice()
115 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
116 if (output_array.data_type == ArrayDataType::kNone) { in Run()
121 if (!output_array.has_shape()) { in Run()
142 CHECK(!output_array.buffer); in Run()
143 switch (output_array.data_type) { in Run()
145 StridedSlice<ArrayDataType::kFloat>(*op, input_array, &output_array); in Run()
[all …]
H A Dresolve_constant_slice.cc28 Array* output_array) { in Slice() argument
32 CHECK(output_array->data_type == Type); in Slice()
37 output_array->GetMutableBuffer<Type>().data; in Slice()
38 output_data.resize(RequiredBufferSizeForShape(output_array->shape())); in Slice()
102 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
103 if (output_array.data_type == ArrayDataType::kNone) { in Run()
108 if (!output_array.has_shape()) { in Run()
128 CHECK(!output_array.buffer); in Run()
129 switch (output_array.data_type) { in Run()
131 if (!Slice<ArrayDataType::kFloat>(*op, input_array, &output_array)) { in Run()
[all …]
H A Dresolve_constant_transpose.cc30 const std::vector<int>& perm, Array* output_array) { in Transpose() argument
35 const Shape& output_shape = output_array->shape(); in Transpose()
37 output_array->GetMutableBuffer<Type>().data; in Transpose()
117 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
118 if (output_array.data_type == ArrayDataType::kNone) { in Run()
122 if (!output_array.has_shape()) { in Run()
134 CopyMinMaxAndQuantizationRelatedFields(input_array, &output_array); in Run()
144 CHECK(!output_array.buffer); in Run()
145 switch (output_array.data_type) { in Run()
148 &output_array); in Run()
[all …]
H A Didentify_nearest_upsample.cc113 Array& output_array = model->GetArray(output); in Run() local
117 !output_array.has_shape()) { in Run()
210 reshape_output_array.data_type = output_array.data_type; in Run()
214 if (output_array.minmax) { in Run()
215 reshape_output_array.GetOrCreateMinMax().min = output_array.minmax->min; in Run()
216 reshape_output_array.GetOrCreateMinMax().max = output_array.minmax->max; in Run()
218 if (output_array.quantization_params) { in Run()
220 output_array.quantization_params->scale; in Run()
222 output_array.quantization_params->zero_point; in Run()
244 if (output_array.minmax) { in Run()
[all …]
H A Dresolve_reorder_axes.cc59 const Array& input_array, Array* output_array) { in ReorderAxes() argument
61 DCHECK(!output_array->buffer); in ReorderAxes()
63 auto& output_data = output_array->GetMutableBuffer<DataType>().data; in ReorderAxes()
64 output_data.resize(RequiredBufferSizeForShape(output_array->shape())); in ReorderAxes()
67 Shape output_shape = output_array->shape(); in ReorderAxes()
75 output_array->GetOrCreateMinMax() = input_array.GetMinMax(); in ReorderAxes()
78 output_array->narrow_range = true; in ReorderAxes()
97 auto& output_array = model->GetArray(output_array_name); in Run() local
102 if (!output_array.has_shape()) { in Run()
109 input_array, &output_array); in Run()
[all …]
H A Dresolve_multiply_by_zero.cc64 auto& output_array = model->GetArray(output_array_name); in Run() local
70 if (output_array.data_type == ArrayDataType::kNone) { in Run()
76 if (!output_array.has_shape()) { in Run()
104 CHECK(constant_input_array.data_type == output_array.data_type); in Run()
105 switch (output_array.data_type) { in Run()
113 FillArrayWithZeros<ArrayDataType::kFloat>(&output_array); in Run()
122 FillArrayWithZeros<ArrayDataType::kUint8>(&output_array); in Run()
131 FillArrayWithZeros<ArrayDataType::kInt32>(&output_array); in Run()
140 FillArrayWithZeros<ArrayDataType::kInt64>(&output_array); in Run()
149 FillArrayWithZeros<ArrayDataType::kComplex64>(&output_array); in Run()
H A Dresolve_constant_pack.cc28 auto& output_array = model->GetArray(op.outputs[0]); in Pack() local
29 CHECK(output_array.data_type == Type); in Pack()
33 output_array.GetMutableBuffer<Type>().data; in Pack()
34 output_data.resize(RequiredBufferSizeForShape(output_array.shape())); in Pack()
65 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
66 if (output_array.data_type == ArrayDataType::kNone) { in Run()
71 if (!output_array.has_shape()) { in Run()
90 CHECK(!output_array.buffer); in Run()
91 switch (output_array.data_type) { in Run()
H A Dresolve_constant_random_uniform.cc35 auto& output_array = model->GetArray(op->outputs[0]); in ComputeRandomUniformArray() local
36 CHECK(output_array.data_type == Type); in ComputeRandomUniformArray()
38 output_array.GetMutableBuffer<Type>().data; in ComputeRandomUniformArray()
39 data.resize(RequiredBufferSizeForShape(output_array.shape())); in ComputeRandomUniformArray()
76 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
77 if (output_array.data_type == ArrayDataType::kNone) { in Run()
82 if (!output_array.has_shape()) { in Run()
95 switch (output_array.data_type) { in Run()
H A Dresolve_constant_fill.cc27 auto& output_array = model->GetArray(op->outputs[0]); in ComputeFillArray() local
30 CHECK(output_array.data_type == Type); in ComputeFillArray()
34 output_array.GetMutableBuffer<Type>().data; in ComputeFillArray()
35 data.resize(RequiredBufferSizeForShape(output_array.shape())); in ComputeFillArray()
58 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
59 if (output_array.data_type == ArrayDataType::kNone) { in Run()
64 if (!output_array.has_shape()) { in Run()
80 switch (output_array.data_type) { in Run()
H A Dresolve_constant_fake_quant.cc98 auto& output_array = model->GetArray(fakequant_op->outputs[0]); in Run() local
100 output_array.data_type = ArrayDataType::kFloat; in Run()
106 output_array.final_data_type = quantized_data_type; in Run()
109 CHECK(!output_array.buffer); in Run()
111 output_array.GetOrCreateMinMax() = *fakequant_op->minmax; in Run()
112 auto& output_buffer = output_array.GetMutableBuffer<ArrayDataType::kFloat>(); in Run()
117 output_array, quantized_data_type, &qparams); in Run()
123 output_array.narrow_range = true; in Run()
H A Dresolve_constant_range.cc25 const Array& delta_array, Array* output_array) { in FillRangeOutput() argument
30 auto& buffer = output_array->GetMutableBuffer<A>(); in FillRangeOutput()
40 CHECK_EQ(buffer.data.size(), output_array->shape().dims()[0]); in FillRangeOutput()
79 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
80 if (output_array.data_type == ArrayDataType::kNone) { in Run()
102 delta_array, &output_array); in Run()
105 delta_array, &output_array); in Run()
H A Dresolve_constant_shape_or_rank.cc33 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
34 if (output_array.data_type == ArrayDataType::kNone) { in Run()
45 if (!output_array.has_shape()) { in Run()
51 CHECK(!output_array.buffer); in Run()
52 auto& output_buffer = output_array.GetMutableBuffer<ArrayDataType::kInt32>(); in Run()
61 output_array.mutable_shape()->ReplaceDims( in Run()
H A Dpropagate_default_min_max.cc60 auto& output_array = model->GetArray(output); in Run() local
61 if (!output_array.minmax && !output_array.buffer && in Run()
62 SupportsMinMax(output_array)) { in Run()
63 did_change |= SetArrayMinMax(output, &output_array); in Run()
H A Dresolve_constant_unary.cc101 auto& output_array = model->GetArray(op.outputs[0]); in CopyMinMaxFromFirstInput() local
102 if (output_array.minmax) { in CopyMinMaxFromFirstInput()
110 CHECK(!output_array.minmax); in CopyMinMaxFromFirstInput()
111 auto& output_minmax = output_array.GetOrCreateMinMax(); in CopyMinMaxFromFirstInput()
159 auto& output_array = model->GetArray(unary_op->outputs[0]); in Run() local
160 if (!output_array.has_shape()) { in Run()
212 const Shape& output_shape = output_array.shape(); in Run()
216 output_array.GetMutableBuffer<ArrayDataType::kFloat>().data; in Run()
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/strings_ops/
H A Dunsorted_segment_join_op_test.py45 output_array = [['Y', '6', '6'], ['Y:p', 'q:G', 'c:a']]
53 self.assertAllEqual(res.shape, np.array(output_array).shape)
54 self.assertAllEqualUnicode(res, output_array)
61 output_array = ['', '', '']
68 self.assertAllEqual(res.shape, np.array(output_array).shape)
69 self.assertAllEqualUnicode(res, output_array)
76 output_array = [['Y', '6', '6'], ['Y:p', 'q:G', 'c:a']]
84 self.assertAllEqual(res.shape, np.array(output_array).shape)
85 self.assertAllEqualUnicode(res, output_array)
95 self.assertAllEqual(res.shape, np.array(output_array).shape)
[all …]
/aosp_15_r20/frameworks/rs/toolkit/
H A DJniEntryPoints.cpp226 jint size_x, jint size_y, jint radius, jbyteArray output_array, jobject restriction) { in Java_android_renderscript_toolkit_Toolkit_nativeBlur() argument
230 ByteArrayGuard output{env, output_array}; in Java_android_renderscript_toolkit_Toolkit_nativeBlur()
249 jint input_vector_size, jint size_x, jint size_y, jbyteArray output_array, in Java_android_renderscript_toolkit_Toolkit_nativeColorMatrix() argument
254 ByteArrayGuard output{env, output_array}; in Java_android_renderscript_toolkit_Toolkit_nativeColorMatrix()
278 jint size_x, jint size_y, jbyteArray output_array, jfloatArray coefficients, in Java_android_renderscript_toolkit_Toolkit_nativeConvolve() argument
283 ByteArrayGuard output{env, output_array}; in Java_android_renderscript_toolkit_Toolkit_nativeConvolve()
321 jint vector_size, jint size_x, jint size_y, jintArray output_array, jobject restriction) { in Java_android_renderscript_toolkit_Toolkit_nativeHistogram() argument
325 IntArrayGuard output{env, output_array}; in Java_android_renderscript_toolkit_Toolkit_nativeHistogram()
332 jintArray output_array, jobject restriction) { in Java_android_renderscript_toolkit_Toolkit_nativeHistogramBitmap() argument
336 IntArrayGuard output{env, output_array}; in Java_android_renderscript_toolkit_Toolkit_nativeHistogramBitmap()
[all …]
/aosp_15_r20/external/renderscript-intrinsics-replacement-toolkit/renderscript-toolkit/src/main/cpp/
H A DJniEntryPoints.cpp226 jint size_x, jint size_y, jint radius, jbyteArray output_array, jobject restriction) { in Java_com_google_android_renderscript_Toolkit_nativeBlur() argument
230 ByteArrayGuard output{env, output_array}; in Java_com_google_android_renderscript_Toolkit_nativeBlur()
249 jint input_vector_size, jint size_x, jint size_y, jbyteArray output_array, in Java_com_google_android_renderscript_Toolkit_nativeColorMatrix() argument
254 ByteArrayGuard output{env, output_array}; in Java_com_google_android_renderscript_Toolkit_nativeColorMatrix()
278 jint size_x, jint size_y, jbyteArray output_array, jfloatArray coefficients, in Java_com_google_android_renderscript_Toolkit_nativeConvolve() argument
283 ByteArrayGuard output{env, output_array}; in Java_com_google_android_renderscript_Toolkit_nativeConvolve()
321 jint vector_size, jint size_x, jint size_y, jintArray output_array, jobject restriction) { in Java_com_google_android_renderscript_Toolkit_nativeHistogram() argument
325 IntArrayGuard output{env, output_array}; in Java_com_google_android_renderscript_Toolkit_nativeHistogram()
332 jintArray output_array, jobject restriction) { in Java_com_google_android_renderscript_Toolkit_nativeHistogramBitmap() argument
336 IntArrayGuard output{env, output_array}; in Java_com_google_android_renderscript_Toolkit_nativeHistogramBitmap()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dtooling_util.cc111 for (const auto& output_array : model.flags.output_arrays()) { in IsOutputArray() local
112 if (array_name == output_array) { in IsOutputArray()
873 for (const std::string& output_array : model_flags.output_arrays()) { in CheckInputArraysAreNotOutputArrays() local
874 QCHECK_NE(input_array.name(), output_array) in CheckInputArraysAreNotOutputArrays()
875 << "The array " << output_array in CheckInputArraysAreNotOutputArrays()
916 for (const std::string& output_array : model_flags.output_arrays()) { in CheckNonAsciiIOArrays() local
917 QCHECK(IsAsciiPrintable(output_array)) in CheckNonAsciiIOArrays()
919 << output_array << ". Pass --allow_nonascii_arrays to allow that. " in CheckNonAsciiIOArrays()
921 << DumpAscii(output_array); in CheckNonAsciiIOArrays()
939 for (const std::string& output_array : model.flags.output_arrays()) { in CheckNonExistentIOArrays() local
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tflite/
H A Dexport_test.cc57 Array& output_array = input_model_.GetOrCreateArray(op->outputs[0]); in AddOperatorsByName() local
60 output_array.data_type = ArrayDataType::kFloat; in AddOperatorsByName()
68 Array& output_array = input_model_.GetOrCreateArray(op->outputs[0]); in AddOperatorsByName() local
71 output_array.data_type = ArrayDataType::kFloat; in AddOperatorsByName()
79 Array& output_array = input_model_.GetOrCreateArray(op->outputs[0]); in AddOperatorsByName() local
82 output_array.data_type = ArrayDataType::kFloat; in AddOperatorsByName()
85 output_array.copy_shape({1, 2, 2, 2}); in AddOperatorsByName()
141 Array& output_array = input_model_.GetOrCreateArray(op->outputs[0]); in BuildQuantizableTestModel() local
144 output_array.data_type = ArrayDataType::kFloat; in BuildQuantizableTestModel()
153 Array& output_array = input_model_.GetOrCreateArray(op->outputs[0]); in BuildQuantizableTestModel() local
[all …]
/aosp_15_r20/external/cronet/base/hash/
H A Dsha1_unittest.cc71 base::SHA1Digest output_array = base::SHA1HashSpan(base::as_byte_span(input)); in TEST() local
73 EXPECT_EQ(kExpected[i], output_array[i]); in TEST()
91 base::SHA1Digest output_array = base::SHA1HashSpan(base::as_byte_span(input)); in TEST() local
93 EXPECT_EQ(kExpected[i], output_array[i]); in TEST()
110 base::SHA1Digest output_array = base::SHA1HashSpan(base::as_byte_span(input)); in TEST() local
112 EXPECT_EQ(kExpected[i], output_array[i]); in TEST()

1234