Home
last modified time | relevance | path

Searched defs:input_array (Results 1 – 25 of 75) 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.cc165 Array4D<float> input_array(1, 0, 2, 1); in XLA_TEST_P() local
178 Array4D<float> input_array(1, 2, 2, 1); in XLA_TEST_P() local
193 Array4D<float> input_array(1, 3, 3, 1); in XLA_TEST_P() local
207 Array4D<float> input_array(3, 6, 7, 32); in XLA_TEST_P() local
224 Array4D<float> input_array(4, 4, 6, 8); in XLA_TEST_P() local
246 Array4D<float> input_array(3, 3, 2, 1); in XLA_TEST_P() local
260 Array4D<float> input_array(9, 12, 4, 89); in XLA_TEST_P() local
285 Array4D<float> input_array(15, 15, 4, 128); in XLA_TEST_P() local
308 Array4D<float> input_array(19, 17, 8, 256); in XLA_TEST_P() local
326 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
224 const auto& input_array = in ProcessTransposeConvOperator() local
255 const auto& input_array = model->GetArray(op->inputs[0]); in ProcessDepthwiseConvOperator() local
297 const auto& input_array = model->GetArray(op->inputs[0]); in ProcessDepthToSpaceOperator() local
320 const auto& input_array = model->GetArray(op->inputs[0]); in ProcessSpaceToDepthOperator() local
370 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_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_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_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.cc40 const Array& input_array, Model* model, in SliceInput()
93 std::vector<int32> GetTransposePerm(const Array& input_array) { in GetTransposePerm()
115 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
Dresolve_constant_shape_or_rank.cc39 const auto& 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
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.cc103 const auto& input_array = model->GetArray(op.inputs[0]); in CopyMinMaxFromFirstInput() local
181 const auto& input_array = model->GetArray(unary_op->inputs[0]); in Run() local
Dpropagate_default_min_max.cc52 auto& input_array = model->GetArray(input); in Run() local
/external/tensorflow/tensorflow/python/ops/ragged/
Dstrings_reduce_join_op_test.py121 input_array, argument
/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.cc63 auto input_array = input.flat<T1>(); in Compute() local

123