Home
last modified time | relevance | path

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

123

/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.cc156 Array4D<float> input_array(1, 0, 2, 1); in XLA_TEST_P() local
169 Array4D<float> input_array(1, 2, 2, 1); in TEST_P() local
184 Array4D<float> input_array(1, 3, 3, 1); in TEST_P() local
198 Array4D<float> input_array(3, 6, 7, 32); in TEST_P() local
215 Array4D<float> input_array(4, 4, 6, 8); in TEST_P() local
237 Array4D<float> input_array(9, 12, 4, 89); in TEST_P() local
262 Array4D<float> input_array(15, 15, 4, 128); in TEST_P() local
285 Array4D<float> input_array(19, 17, 8, 256); in TEST_P() local
303 Array4D<float> input_array(1, 2, 2, 1); in XLA_TEST_P() local
340 Array4D<float> input_array(13, 12, 8, 15); in TEST_P() local
[all …]
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc125 const auto& input_array = model->GetArray(op->inputs[0]); in ProcessConvOperator() local
223 const auto& input_array = in ProcessTransposeConvOperator() local
254 const auto& input_array = model->GetArray(op->inputs[0]); in ProcessDepthwiseConvOperator() local
296 const auto& input_array = model->GetArray(op->inputs[0]); in ProcessDepthToSpaceOperator() local
319 const auto& input_array = model->GetArray(op->inputs[0]); in ProcessSpaceToDepthOperator() local
369 const auto& input_array = model->GetArray(op->inputs[0]); in ProcessFullyConnectedOperator() local
403 const auto& input_array = model->GetArray(op->inputs[0]); in ProcessTensorFlowReshapeOperator() local
466 const auto& input_array = model->GetArray(op->inputs[input_index]); in ProcessSimpleOperator() local
498 const auto& input_array = model->GetArray(input); in ProcessSelectOperator() local
519 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
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_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.cc58 const Array& input_array, Array* output_array) { in ReorderAxes()
95 auto& input_array = model->GetArray(input_array_name); 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_tile.cc72 inline void Tile(const Array& input_array, const Array& multiples_array, in Tile()
128 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
Dresolve_constant_reshape.cc55 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
Dremove_trivial_concatenation_input.cc47 const auto& input_array = model->GetArray(input); in Run() 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
Dremove_trivial_slice.cc36 const auto& input_array = model.GetArray(op.inputs[0]); in IsSliceTrivial() local
Dconvert_trivial_pack_to_reshape.cc43 const auto& input_array = model->GetArray(pack_op->inputs[0]); in Run() local
Dremove_trivial_reshape.cc37 const auto& input_array = model.GetArray(op.inputs[0]); in IsReshapeTrivial() 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
Dpropagate_default_min_max.cc52 auto& input_array = model->GetArray(input); in Run() local
Dremove_trivial_quantized_min_max.cc63 const auto& input_array = model.GetArray(input_array_name); in IsTrivialMinMax() local
Dmake_initial_dequantize_operator.cc55 auto& input_array = model->GetArray(input_name); in AddDequantizeOperatorToInput() local
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc68 const Array& input_array = op_signature.model->GetArray(input_name); in GetVersion() local
109 const Array& input_array = op_signature.model->GetArray(input_name); in GetVersion() local
166 const Array& input_array = op_signature.model->GetArray(input_name); in GetVersion() local
204 const Array& input_array = op_signature.model->GetArray(input_name); in GetVersion() local
250 const Array& input_array = op_signature.model->GetArray(input_name); in GetVersion() local
280 const Array& input_array = op_signature.model->GetArray(input_name); in GetVersion() local
331 const Array& input_array = op_signature.model->GetArray(input_name); in GetVersion() local
382 const Array& input_array = op_signature.model->GetArray(input_name); in GetVersion() local
494 const Array& input_array = op_signature.model->GetArray(input_name); in GetVersion() local
540 const Array& input_array = op_signature.model->GetArray(input_name); in GetVersion() local
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dreduce_join_op_test.py101 input_array, argument
128 def _testMultipleReduceJoin(self, input_array, axis, separator=" "): argument
/external/tensorflow/tensorflow/core/kernels/
Drequantize.cc64 auto input_array = input.flat<T1>(); in Compute() local

123