Home
last modified time | relevance | path

Searched defs:input_array (Results 1 – 25 of 79) sorted by relevance

1234

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dconvolution_variants_test.cc56 const Array4D<float> input_array(1, 1, 1, 1, {2}); in XLA_TEST_F() local
71 const Array4D<float> input_array(5, 1, 1, 1, {1, 2, 3, 4, 5}); in XLA_TEST_F() local
86 Array4D<float> input_array(2, 1, 3, 4); in XLA_TEST_F() local
103 Array4D<float> input_array(1, 2, 1, 1, {10, 1}); in XLA_TEST_F() local
118 Array4D<float> input_array(1, 1, 1, 2, {1, 2}); in XLA_TEST_F() local
133 Array4D<float> input_array(1, 1, 1, 3, {1, 2, 3}); in XLA_TEST_F() local
148 Array4D<float> input_array(1, 1, 2, 2, {1, 2, 3, 4}); in XLA_TEST_F() local
163 Array4D<float> input_array(1, 1, 2, 2, {1, 2, 3, 4}); in XLA_TEST_F() local
178 Array4D<float> input_array(1, 1, 2, 2, {1, 2, 3, 4}); in XLA_TEST_F() local
193 Array4D<float> input_array( in XLA_TEST_F() local
[all …]
H A Dreduce_window_test.cc163 Array4D<float> input_array(1, 0, 2, 1); in XLA_TEST_P() local
176 Array4D<float> input_array(1, 2, 2, 1); in XLA_TEST_P() local
191 Array4D<float> input_array(1, 3, 3, 1); in XLA_TEST_P() local
205 Array4D<float> input_array(3, 6, 7, 32); in XLA_TEST_P() local
222 Array4D<float> input_array(4, 4, 6, 8); in XLA_TEST_P() local
244 Array4D<float> input_array(3, 3, 2, 1); in XLA_TEST_P() local
258 Array4D<float> input_array(9, 12, 4, 89); in XLA_TEST_P() local
283 Array4D<float> input_array(15, 15, 4, 128); in XLA_TEST_P() local
306 Array4D<float> input_array(19, 17, 8, 256); in XLA_TEST_P() local
324 Array4D<float> input_array(1, 2, 2, 1); in XLA_TEST_P() local
[all …]
H A Dreshape_test.cc55 Array2D<float> input_array(1, 1); in XLA_TEST_P() local
100 Array2D<float> input_array(1, 1); in XLA_TEST_P() local
134 Array2D<float> input_array(0, 3); in XLA_TEST_P() local
163 Array2D<float> input_array(3, 0); in XLA_TEST_P() local
792 Array4D<float> input_array(2, 3, 5, 7); in XLA_TEST_P() local
H A Dpad_test.cc207 Array4D<float> input_array(1, 1, 2, 3, {1, 2, 3, 4, 5, 6}); in TEST_P() local
249 Array4D<float> input_array(1, 25, 7, 7); in XLA_TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dpropagate_fixed_sizes.cc126 const auto& input_array = model->GetArray(op->inputs[0]); in ProcessConvOperator() local
222 const auto& input_array = in ProcessTransposeConvOperator() local
253 const auto& input_array = model->GetArray(op->inputs[0]); in ProcessDepthwiseConvOperator() local
295 const auto& input_array = model->GetArray(op->inputs[0]); in ProcessDepthToSpaceOperator() local
318 const auto& input_array = model->GetArray(op->inputs[0]); in ProcessSpaceToDepthOperator() local
368 const auto& input_array = model->GetArray(op->inputs[0]); in ProcessFullyConnectedOperator() local
404 const auto& input_array = model->GetArray(op->inputs[0]); in ProcessTensorFlowReshapeOperator() local
467 const auto& input_array = model->GetArray(op->inputs[input_index]); in ProcessSimpleOperator() local
499 const auto& input_array = model->GetArray(input); in ProcessSelectOperator() local
520 const auto& input_array = model->GetArray(input); in ProcessAddNOperator() local
[all …]
H A Dhardcode_min_max.cc37 const auto& input_array = model->GetArray(op->inputs[0]); in HardcodeMinMaxForIm2colArray() local
54 const auto& input_array = model->GetArray(op->inputs[0]); in HardcodeMinMaxForL2Normalization() local
156 auto& input_array = model->GetArray(op->inputs[1]); in HardcodeMinMaxForSplit() local
177 const auto& input_array = model->GetArray(op->inputs[0]); in HardcodeMinMaxForAverageOrMaxPool() local
194 const auto& input_array = model->GetArray(op->inputs[0]); in HardcodeMinMaxFromFirstInput() local
259 const auto& input_array = model->GetArray(op->inputs[0]); in HardcodeMinMaxForOutput() local
409 const auto& input_array = model->GetArray(op->inputs[i]); in HardcodeMinMaxForPack() local
H A Dunroll_batch_matmul.cc41 const Array& input_array, Model* model, in SliceInput()
95 std::vector<int32> GetTransposePerm(const Array& input_array) { in GetTransposePerm()
117 const auto& input_array = model->GetArray(input); in TransposeInput() local
H A Dresolve_constant_strided_slice.cc28 void StridedSlice(StridedSliceOperator const& op, Array const& input_array, in StridedSlice()
132 const auto& input_array = model->GetArray(op->inputs[0]); in Run() local
H A Dresolve_constant_gather.cc28 inline void Gather(const Array& input_array, const Array& coords_array, in Gather()
106 const Array& input_array = model->GetArray(op->inputs[0]); in Run() local
H A Dresolve_reorder_axes.cc59 const Array& input_array, Array* output_array) { in ReorderAxes()
96 auto& input_array = model->GetArray(input_array_name); in Run() local
H A Dresolve_constant_slice.cc27 bool Slice(SliceOperator const& op, Array const& input_array, in Slice()
118 const auto& input_array = model->GetArray(op->inputs[0]); in Run() local
H A Dresolve_constant_tile.cc73 inline void Tile(const Array& input_array, const Array& multiples_array, in Tile()
129 const Array& input_array = model->GetArray(op->inputs[0]); in Run() local
H A Dresolve_constant_transpose.cc29 void Transpose(Model* model, const Array& input_array, in Transpose()
132 const Array& input_array = model->GetArray(op->inputs[0]); in Run() local
H A Dresolve_constant_unary.cc105 const auto& input_array = model->GetArray(op.inputs[0]); in CopyMinMaxFromFirstInput() local
183 const auto& input_array = model->GetArray(unary_op->inputs[0]); in Run() local
H A Dquantize.cc425 const auto& input_array = model->GetArray(op.inputs[data_input_index]); in ChooseQuantizationForOperatorOutput() local
536 const auto& input_array = model->GetArray(input); in Run() local
H A Dresolve_constant_shape_or_rank.cc39 const auto& input_array = model->GetArray(op->inputs[0]); in Run() local
H A Dresolve_constant_reshape.cc55 const Array& input_array = model->GetArray(op->inputs[0]); in Run() local
H A Dconvert_squeeze_to_reshape.cc45 const auto& input_array = model->GetArray(squeeze_op->inputs[0]); in Run() local
/aosp_15_r20/frameworks/rs/toolkit/
H A DJniEntryPoints.cpp225 JNIEnv* env, jobject /*thiz*/, jlong native_handle, jbyteArray input_array, jint vectorSize, in Java_android_renderscript_toolkit_Toolkit_nativeBlur()
248 JNIEnv* env, jobject /*thiz*/, jlong native_handle, jbyteArray input_array, in Java_android_renderscript_toolkit_Toolkit_nativeColorMatrix()
277 JNIEnv* env, jobject /*thiz*/, jlong native_handle, jbyteArray input_array, jint vectorSize, in Java_android_renderscript_toolkit_Toolkit_nativeConvolve()
320 JNIEnv* env, jobject /*thiz*/, jlong native_handle, jbyteArray input_array, in Java_android_renderscript_toolkit_Toolkit_nativeHistogram()
343 JNIEnv* env, jobject /*thiz*/, jlong native_handle, jbyteArray input_array, in Java_android_renderscript_toolkit_Toolkit_nativeHistogramDot()
371 JNIEnv* env, jobject /*thiz*/, jlong native_handle, jbyteArray input_array, in Java_android_renderscript_toolkit_Toolkit_nativeLut()
408 JNIEnv* env, jobject /*thiz*/, jlong native_handle, jbyteArray input_array, in Java_android_renderscript_toolkit_Toolkit_nativeLut3d()
436 JNIEnv* env, jobject /*thiz*/, jlong native_handle, jbyteArray input_array, in Java_android_renderscript_toolkit_Toolkit_nativeResize()
461 JNIEnv* env, jobject /*thiz*/, jlong native_handle, jbyteArray input_array, in Java_android_renderscript_toolkit_Toolkit_nativeYuvToRgb()
472 JNIEnv* env, jobject /*thiz*/, jlong native_handle, jbyteArray input_array, jint size_x, in Java_android_renderscript_toolkit_Toolkit_nativeYuvToRgbBitmap()
/aosp_15_r20/external/renderscript-intrinsics-replacement-toolkit/renderscript-toolkit/src/main/cpp/
H A DJniEntryPoints.cpp225 JNIEnv* env, jobject /*thiz*/, jlong native_handle, jbyteArray input_array, jint vectorSize, in Java_com_google_android_renderscript_Toolkit_nativeBlur()
248 JNIEnv* env, jobject /*thiz*/, jlong native_handle, jbyteArray input_array, in Java_com_google_android_renderscript_Toolkit_nativeColorMatrix()
277 JNIEnv* env, jobject /*thiz*/, jlong native_handle, jbyteArray input_array, jint vectorSize, in Java_com_google_android_renderscript_Toolkit_nativeConvolve()
320 JNIEnv* env, jobject /*thiz*/, jlong native_handle, jbyteArray input_array, in Java_com_google_android_renderscript_Toolkit_nativeHistogram()
343 JNIEnv* env, jobject /*thiz*/, jlong native_handle, jbyteArray input_array, in Java_com_google_android_renderscript_Toolkit_nativeHistogramDot()
371 JNIEnv* env, jobject /*thiz*/, jlong native_handle, jbyteArray input_array, in Java_com_google_android_renderscript_Toolkit_nativeLut()
408 JNIEnv* env, jobject /*thiz*/, jlong native_handle, jbyteArray input_array, in Java_com_google_android_renderscript_Toolkit_nativeLut3d()
436 JNIEnv* env, jobject /*thiz*/, jlong native_handle, jbyteArray input_array, in Java_com_google_android_renderscript_Toolkit_nativeResize()
461 JNIEnv* env, jobject /*thiz*/, jlong native_handle, jbyteArray input_array, in Java_com_google_android_renderscript_Toolkit_nativeYuvToRgb()
472 JNIEnv* env, jobject /*thiz*/, jlong native_handle, jbyteArray input_array, jint size_x, in Java_com_google_android_renderscript_Toolkit_nativeYuvToRgbBitmap()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dtooling_util.cc102 for (const auto& input_array : model.flags.input_arrays()) { in IsInputArray() local
872 for (const auto& input_array : model_flags.input_arrays()) { in CheckInputArraysAreNotOutputArrays() local
908 for (const auto& input_array : model_flags.input_arrays()) { in CheckNonAsciiIOArrays() local
1494 for (const auto& input_array : model->flags.input_arrays()) { in CreateOrCheckRnnStateArray() local
1636 auto& input_array = model->GetOrCreateArray(input_array_proto.name()); in ResolveModelFlags() local
1721 const auto& input_array = model.GetArray(input); in CheckIsReadyForQuantization() local
1924 const auto& input_array = model.GetArray(op.inputs[2]); in EstimateArithmeticOpsCount() local
2256 const auto& input_array = model.GetArray(op->inputs[0]); in ReshapeIsEquivalentToTranspose() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dquantization_utils_test.cc53 auto input_array = i_tensor.flat<qint32>(); in TestRequantizeMany() local
94 const auto input_array = i_tensor.flat<quint8>(); in TestRequantizeMany8To32Bit() local
354 auto input_array = input.flat<float>(); in TestFloatToQuantizedInPlaceUsingEigen() local
400 auto input_array = input.flat<T>(); in TestQuantizedToFloatInPlaceUsingEigen() local
670 auto input_array = input.flat<float>(); in TestOverflowWithEigen() local
H A Dquantization_utils.h150 #define DEQUANTIZE_WITH_EIGEN(input_array, q2f) \ argument
158 #define QUANTIZE_WITH_EIGEN(input_array, f2q, OutputType) \ argument
680 auto input_array = input.flat<T1>(); in RequantizeManyInNewRangeUsingEigen() local
721 auto input_array = input.flat<qint32>(); variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tflite/
H A Dexport_test.cc55 Array& input_array = input_model_.GetOrCreateArray(op->inputs[0]); in AddOperatorsByName() local
139 Array& input_array = input_model_.GetArray(op->inputs[0]); in BuildQuantizableTestModel() local
687 Array& input_array = model.GetOrCreateArray(op->inputs[0]); in TEST() local
709 Array& input_array = model.GetOrCreateArray(op->inputs[0]); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/strings_ops/
H A Dreduce_join_op_test.py97 input_array, argument
124 def _testMultipleReduceJoin(self, input_array, axis, separator=" "): argument

1234