Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_variants_test.cc57 const Array4D<float> input_array(1, 1, 1, 1, {2}); in XLA_TEST_F() local
72 const Array4D<float> input_array(5, 1, 1, 1, {1, 2, 3, 4, 5}); in XLA_TEST_F() local
87 Array4D<float> input_array(2, 1, 3, 4); in XLA_TEST_F() local
104 Array4D<float> input_array(1, 2, 1, 1, {10, 1}); in XLA_TEST_F() local
119 Array4D<float> input_array(1, 1, 1, 2, {1, 2}); in XLA_TEST_F() local
134 Array4D<float> input_array(1, 1, 1, 3, {1, 2, 3}); in XLA_TEST_F() local
149 Array4D<float> input_array(1, 1, 2, 2, {1, 2, 3, 4}); in XLA_TEST_F() local
164 Array4D<float> input_array(1, 1, 2, 2, {1, 2, 3, 4}); in XLA_TEST_F() local
179 Array4D<float> input_array(1, 1, 2, 2, {1, 2, 3, 4}); in XLA_TEST_F() local
194 Array4D<float> input_array( in XLA_TEST_F() local
[all …]
Dreduce_window_test.cc152 Array4D<float> input_array(1, 0, 2, 1); in XLA_TEST_P() local
165 Array4D<float> input_array(1, 2, 2, 1); in TEST_P() local
180 Array4D<float> input_array(1, 3, 3, 1); in TEST_P() local
194 Array4D<float> input_array(3, 6, 7, 32); in TEST_P() local
211 Array4D<float> input_array(4, 4, 6, 8); in TEST_P() local
233 Array4D<float> input_array(9, 12, 4, 89); in TEST_P() local
257 Array4D<float> input_array(1, 2, 2, 1); in XLA_TEST_P() local
294 Array4D<float> input_array(13, 12, 8, 15); in TEST_P() local
362 Array4D<float> input_array(2, 1, 27, 119); in XLA_TEST_P() local
384 Array4D<float> input_array(3, 2, 4, 64); in XLA_TEST_P() local
[all …]
Dconvolution_dimension_numbers_test.cc71 auto input_array = MakeUnique<Array4D<float>>(2, 3, 5, 5); in XLA_TEST_F() local
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc148 const auto& input_array = model->GetArray(op->inputs[0]); in ProcessConvOperator() local
190 const auto& input_array = model->GetArray(op->inputs[0]); in ProcessDepthwiseConvOperator() local
231 const auto& input_array = model->GetArray(op->inputs[0]); in ProcessDepthToSpaceOperator() local
254 const auto& input_array = model->GetArray(op->inputs[0]); in ProcessSpaceToDepthOperator() local
309 const auto& input_array = model->GetArray(op->inputs[0]); in ProcessFullyConnectedOperator() local
343 const auto& input_array = model->GetArray(op->inputs[0]); in ProcessTensorFlowReshapeOperator() local
399 const auto& input_array = model->GetArray(op->inputs[0]); in ProcessSimpleOperator() local
437 const auto& input_array = model->GetArray(input); in ProcessAddNOperator() local
472 const auto& input_array = model->GetArray(op->inputs[0]); in ProcessTensorFlowReductionOperator() local
520 const auto& input_array = model->GetArray(op->inputs[0]); in ProcessSliceOperator() 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
134 const auto& input_array = model->GetArray(op->inputs[0]); in HardcodeMinMaxForAverageOrMaxPool() local
151 const auto& input_array = model->GetArray(op->inputs[0]); in HardcodeMinMaxFromFirstInput() local
170 const auto& input_array = model->GetArray(op->inputs[0]); in HardcodeMinMaxForOutput() local
Dresolve_reorder_axes.cc34 Array* input_array, Array* output_array) { in ReorderAxes()
61 auto& input_array = model->GetArray(input_array_name); in Run() local
Dresolve_constant_strided_slice.cc61 void StridedSlice(StridedSliceOperator const& op, Array const& input_array, in StridedSlice()
155 const auto& input_array = model->GetArray(op->inputs[0]); in Run() local
Dresolve_constant_shape_or_rank.cc37 const auto& input_array = model->GetArray(op->inputs[0]); in Run() local
Dremove_trivial_concatenation_input.cc44 const auto& input_array = model->GetArray(input); in Run() local
Dconvert_trivial_stack_to_reshape.cc40 const auto& input_array = model->GetArray(stack_op->inputs[0]); 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
Dresolve_constant_transpose.cc29 void Transpose(Model* model, const Array& input_array, in Transpose()
129 const Array& input_array = model->GetArray(op->inputs[0]); in Run() local
Dconvert_expanddims_to_reshape.cc38 const auto& input_array = model->GetArray(expand_op->inputs[0]); in Run() local
Dresolve_constant_fake_quant.cc47 const auto& input_array = model->GetArray(fakequant_op->inputs[0]); in Run() local
Dresolve_constant_stack.cc41 const auto& input_array = model->GetArray(op.inputs[i]); in Stack() local
Dmake_initial_dequantize_operator.cc54 auto& input_array = model->GetArray(input_name); in AddDequantizeOperatorToInput() local
Didentify_l2_normalization.cc84 const auto& input_array = in Run() local
Dresolve_constant_unary.cc79 const auto& input_array = model->GetArray(unary_op->inputs[0]); in Run() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dreduce_join_op_test.py100 input_array, argument
128 input_array, argument
/external/tensorflow/tensorflow/core/kernels/
Drequantize.cc64 auto input_array = input.flat<T1>(); in Compute() local
Dquantize_down_and_shrink_range.cc53 auto input_array = input.flat<T1>(); in Compute() local
Drequantization_range_op.cc36 auto input_array = input.flat<qint32>(); in CalculateUsedRange() local
/external/libjpeg-turbo/
Djutils.c94 jcopy_sample_rows (JSAMPARRAY input_array, int source_row, in jcopy_sample_rows()
/external/tensorflow/tensorflow/contrib/lite/toco/
Dtooling_util.cc66 for (const auto& input_array : model.flags.input_arrays()) { in IsInputArray() local
646 for (const auto& input_array : model_flags.input_arrays()) { in CheckInputArraysAreNotOutputArrays() local
682 for (const auto& input_array : model_flags.input_arrays()) { in CheckNonAsciiIOArrays() local
703 for (const auto& input_array : model.flags.input_arrays()) { in CheckNonExistentIOArrays() local
1064 for (const auto& input_array : model->flags.input_arrays()) { in CreateOrCheckRnnStateArray() local
1205 auto& input_array = model->GetOrCreateArray(input_array_proto.name()); in ResolveModelFlags() local
1274 for (const auto& input_array : model->flags.input_arrays()) { in ResolveModelFlags() local
1291 const auto& input_array = model.GetArray(input); in CheckIsReadyForQuantization() local
1323 auto& input_array = model->GetArray(input); in UseDefaultMinMaxRangeValues() local
1772 for (const auto& input_array : model.flags.input_arrays()) { in IsDiscardableArray() local

12