Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/compiler/xla/tests/
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 …]
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 …]
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
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 …]
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
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
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
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
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
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
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
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
Dconvert_squeeze_to_reshape.cc45 const auto& input_array = model->GetArray(squeeze_op->inputs[0]); in Run() local
Dresolve_constant_reshape.cc55 const Array& input_array = model->GetArray(op->inputs[0]); in Run() local
Dresolve_constant_shape_or_rank.cc39 const auto& input_array = model->GetArray(op->inputs[0]); in Run() local
Dremove_trivial_concatenation_input.cc47 const auto& input_array = model->GetArray(input); in Run() local
Dremove_trivial_slice.cc36 const auto& input_array = model.GetArray(op.inputs[0]); in IsSliceTrivial() local
Dremove_trivial_reshape.cc37 const auto& input_array = model.GetArray(op.inputs[0]); in IsReshapeTrivial() local
Dconvert_trivial_pack_to_reshape.cc43 const auto& input_array = model->GetArray(pack_op->inputs[0]); in Run() local
Dconvert_trivial_tile_to_concat.cc34 const auto& input_array = model->GetArray(tile_op->inputs[0]); in Run() local
Dconvert_expanddims_to_reshape.cc41 const auto& input_array = model->GetArray(expand_op->inputs[0]); in Run() local
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
Dremove_trivial_quantized_min_max.cc63 const auto& input_array = model.GetArray(input_array_name); in IsTrivialMinMax() local
/external/renderscript-intrinsics-replacement-toolkit/renderscript-toolkit/src/main/cpp/
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()
/external/tensorflow/tensorflow/python/ops/ragged/
Dstrings_reduce_join_op_test.py117 input_array, argument
/external/tensorflow/tensorflow/python/kernel_tests/strings_ops/
Dreduce_join_op_test.py97 input_array, argument
124 def _testMultipleReduceJoin(self, input_array, axis, separator=" "): argument

1234