Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc85 Array* output_array) { in ComputeBinaryOperatorOutputSize()
143 auto& output_array = model->GetArray(op->outputs[0]); in ProcessConvOperator() local
240 auto& output_array = model->GetArray(op->outputs[0]); in ProcessTransposeConvOperator() local
344 auto& output_array = model->GetArray(op->outputs[0]); in ProcessOpWithShapeInput() local
391 auto& output_array = model->GetArray(op->outputs[0]); in ProcessFullyConnectedOperator() local
397 auto& output_array = model->GetArray(op->outputs[0]); in ProcessTensorFlowReshapeOperator() local
473 auto& output_array = model->GetArray(output_name); in ProcessSimpleOperator() local
490 auto& output_array = model->GetArray(output_name); in ProcessSimpleBinaryOperator() local
506 auto& output_array = model->GetArray(op->outputs[0]); in ProcessSelectOperator() local
528 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
Dresolve_constant_range.cc24 const Array& delta_array, Array* output_array) { in FillRangeOutput()
78 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_constant_slice.cc28 Array* output_array) { in Slice()
102 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
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
Dresolve_constant_tile.cc73 Array* output_array) { in Tile()
113 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
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
Dunfuse_activation_functions.cc71 const auto& output_array = model->GetArray(op->outputs[0]); in Run() local
Ddequantize.cc136 for (const string& output_array : model->flags.output_arrays()) { in DequantizeArray() local
205 auto& output_array = model->GetArray(op->outputs[0]); in Run() local
Dresolve_constant_select.cc43 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
Dremove_trivial_slice.cc37 const auto& output_array = model.GetArray(op.outputs[0]); in IsSliceTrivial() local
Dremove_unused_op.cc61 for (const string& output_array : model->flags.output_arrays()) { in Run() local
Dremove_trivial_reshape.cc38 const auto& output_array = model.GetArray(op.outputs[0]); in IsReshapeTrivial() local
Dconvert_trivial_tile_to_concat.cc36 const auto& output_array = model->GetArray(tile_op->outputs[0]); in Run() local
Dresolve_constant_binary.cc73 auto& output_array = model->GetArray(output_name); in EvaluateBinaryOperatorOnConstantInputs() local
221 auto& output_array = model->GetArray(binary_op->outputs[0]); in Run() local
/external/tensorflow/tensorflow/lite/toco/tflite/
Dexport_test.cc48 Array& output_array = input_model_.GetOrCreateArray(op->outputs[0]); in AddOperatorsByName() local
59 Array& output_array = input_model_.GetOrCreateArray(op->outputs[0]); in AddOperatorsByName() local
70 Array& output_array = input_model_.GetOrCreateArray(op->outputs[0]); in AddOperatorsByName() local
129 Array& output_array = input_model_.GetOrCreateArray(op->outputs[0]); in BuildQuantizableTestModel() local
141 Array& output_array = input_model_.GetOrCreateArray(op->outputs[0]); in BuildQuantizableTestModel() local
504 Array& output_array = model.GetOrCreateArray(op->outputs[0]); in TEST() local
525 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
843 for (const string& output_array : model_flags.output_arrays()) { in CheckInputArraysAreNotOutputArrays() local
886 for (const string& output_array : model_flags.output_arrays()) { in CheckNonAsciiIOArrays() local
908 for (const string& output_array : model.flags.output_arrays()) { in CheckNonExistentIOArrays() local
960 for (const string& output_array : model->flags.output_arrays()) { in FixNoMissingArray() local
1077 for (const string& output_array : model.flags.output_arrays()) { in CheckOperatorOrdering() local
1879 const auto& output_array = model.GetArray(op.outputs[0]); in EstimateArithmeticOpsCount() local
1900 const auto& output_array = model.GetArray(op.outputs[0]); in EstimateArithmeticOpsCount() local
1908 const auto& output_array = model.GetArray(op.outputs[0]); in EstimateArithmeticOpsCount() local
1921 const auto& output_array = model.GetArray(op.outputs[0]); in EstimateArithmeticOpsCount() local
[all …]

123