Home
last modified time | relevance | path

Searched defs:output_array (Results 1 – 25 of 57) sorted by relevance

123

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc86 Array* output_array) { in ComputeBinaryOperatorOutputSize()
144 auto& output_array = model->GetArray(op->outputs[0]); in ProcessConvOperator() local
241 auto& output_array = model->GetArray(op->outputs[0]); in ProcessTransposeConvOperator() local
345 auto& output_array = model->GetArray(op->outputs[0]); in ProcessOpWithShapeInput() local
392 auto& output_array = model->GetArray(op->outputs[0]); in ProcessFullyConnectedOperator() local
398 auto& output_array = model->GetArray(op->outputs[0]); in ProcessTensorFlowReshapeOperator() local
474 auto& output_array = model->GetArray(output_name); in ProcessSimpleOperator() local
491 auto& output_array = model->GetArray(output_name); in ProcessSimpleBinaryOperator() local
507 auto& output_array = model->GetArray(op->outputs[0]); in ProcessSelectOperator() local
529 auto& output_array = model->GetArray(op->outputs[0]); in ProcessAddNOperator() local
[all …]
Dresolve_constant_fill.cc27 auto& output_array = model->GetArray(op->outputs[0]); in ComputeFillArray() local
58 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
Dresolve_constant_random_uniform.cc35 auto& output_array = model->GetArray(op->outputs[0]); in ComputeRandomUniformArray() local
76 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
Dresolve_constant_pack.cc28 auto& output_array = model->GetArray(op.outputs[0]); in Pack() local
65 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
Dhardcode_min_max.cc50 auto& output_array = model->GetArray(op->outputs[0]); in HardcodeMinMaxForL2Normalization() local
173 auto& output_array = model->GetArray(op->outputs[0]); in HardcodeMinMaxForAverageOrMaxPool() local
190 auto& output_array = model->GetArray(op->outputs[0]); in HardcodeMinMaxFromFirstInput() local
207 auto& output_array = model->GetArray(op->outputs[0]); in HardcodeMinMaxForSelect() local
255 auto& output_array = model->GetArray(op->outputs[0]); in HardcodeMinMaxForOutput() local
395 auto& output_array = model->GetArray(op->outputs[0]); in HardcodeMinMaxForPack() local
Dresolve_constant_range.cc25 const Array& delta_array, Array* output_array) { in FillRangeOutput()
79 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
Dresolve_constant_strided_slice.cc29 Array* output_array) { in StridedSlice()
115 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
Dresolve_reorder_axes.cc58 const Array& input_array, Array* output_array) { in ReorderAxes()
96 auto& output_array = model->GetArray(output_array_name); in Run() local
Dresolve_constant_gather.cc29 Array* output_array) { in Gather()
80 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
Dresolve_constant_slice.cc28 Array* output_array) { in Slice()
102 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
Dresolve_constant_tile.cc74 Array* output_array) { in Tile()
114 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
Ddequantize.cc136 for (const std::string& output_array : model->flags.output_arrays()) { in DequantizeArray() local
205 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
Dresolve_constant_transpose.cc30 const std::vector<int>& perm, Array* output_array) { in Transpose()
117 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
Dresolve_constant_shape_or_rank.cc33 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
Dresolve_constant_reshape.cc45 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
Ddrop_fake_quant.cc42 const auto& output_array = model->GetArray(fakequant_op->outputs[0]); in Run() local
Dunfuse_activation_functions.cc71 const auto& output_array = model->GetArray(op->outputs[0]); in Run() local
Dremove_unused_op.cc61 for (const std::string& output_array : model->flags.output_arrays()) { in Run() local
Dresolve_constant_select.cc43 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
Dremove_trivial_slice.cc37 const auto& output_array = model.GetArray(op.outputs[0]); in IsSliceTrivial() local
Dresolve_constant_binary.cc74 auto& output_array = model->GetArray(output_name); in EvaluateBinaryOperatorOnConstantInputs() local
221 auto& output_array = model->GetArray(binary_op->outputs[0]); in Run() local
Dremove_trivial_reshape.cc38 const auto& output_array = model.GetArray(op.outputs[0]); in IsReshapeTrivial() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dunsorted_segment_join_op_test.py192 output_array): argument
216 def testSeparator(self, separator, output_array): argument
/external/tensorflow/tensorflow/lite/toco/tflite/
Dexport_test.cc52 Array& output_array = input_model_.GetOrCreateArray(op->outputs[0]); in AddOperatorsByName() local
63 Array& output_array = input_model_.GetOrCreateArray(op->outputs[0]); in AddOperatorsByName() local
74 Array& output_array = input_model_.GetOrCreateArray(op->outputs[0]); in AddOperatorsByName() local
136 Array& output_array = input_model_.GetOrCreateArray(op->outputs[0]); in BuildQuantizableTestModel() local
148 Array& output_array = input_model_.GetOrCreateArray(op->outputs[0]); in BuildQuantizableTestModel() local
688 Array& output_array = model.GetOrCreateArray(op->outputs[0]); in TEST() local
709 Array& output_array = model.GetOrCreateArray(op->outputs[0]); in TEST() local
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc109 for (const auto& output_array : model.flags.output_arrays()) { in IsOutputArray() local
871 for (const std::string& output_array : model_flags.output_arrays()) { in CheckInputArraysAreNotOutputArrays() local
914 for (const std::string& output_array : model_flags.output_arrays()) { in CheckNonAsciiIOArrays() local
937 for (const std::string& output_array : model.flags.output_arrays()) { in CheckNonExistentIOArrays() local
989 for (const std::string& output_array : model->flags.output_arrays()) { in FixNoMissingArray() local
1108 for (const std::string& output_array : model.flags.output_arrays()) { in CheckOperatorOrdering() local
1903 const auto& output_array = model.GetArray(op.outputs[0]); in EstimateArithmeticOpsCount() local
1944 const auto& output_array = model.GetArray(op.outputs[0]); in EstimateArithmeticOpsCount() local
1952 const auto& output_array = model.GetArray(op.outputs[0]); in EstimateArithmeticOpsCount() local
1965 const auto& output_array = model.GetArray(op.outputs[0]); in EstimateArithmeticOpsCount() local
[all …]

123