| /external/libtextclassifier/utils/tflite/ |
| D | text_encoder_test.cc | 93 output_encoding_ = AddOutput(tflite::TensorType_INT32); in TextEncoderOpModel()
94 output_positions_ = AddOutput(tflite::TensorType_INT32); in TextEncoderOpModel()
95 output_length_ = AddOutput(tflite::TensorType_INT32); in TextEncoderOpModel()
96 output_attributes_int32_ = AddOutput(tflite::TensorType_INT32); in TextEncoderOpModel()
97 output_attributes_float_ = AddOutput(tflite::TensorType_FLOAT32); in TextEncoderOpModel()
|
| D | token_encoder_test.cc | 80 output_positions_ = AddOutput(tflite::TensorType_INT32); in TokenEncoderOpModel()
81 output_length_ = AddOutput(tflite::TensorType_INT32); in TokenEncoderOpModel()
82 output_attributes_int32_ = AddOutput(tflite::TensorType_INT32); in TokenEncoderOpModel()
83 output_attributes_float_ = AddOutput(tflite::TensorType_FLOAT32); in TokenEncoderOpModel()
|
| D | dist_diversification_test.cc | 60 output_indexes_ = AddOutput(tflite::TensorType_INT32); in DistanceDiversificationOpModel()
61 output_len_ = AddOutput(tflite::TensorType_INT32); in DistanceDiversificationOpModel()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 97 output_ = AddOutput(output); in FloatAddOpModel()
168 output_ = AddOutput(output); in FloatMulOpModel()
202 output_ = AddOutput(output); in FloatPoolingOpModel()
287 output_ = AddOutput(output); in ConvolutionOpModel()
466 output_ = AddOutput(output); in DepthwiseConvolutionOpModel()
547 output_ = AddOutput(output); in FullyConnectedOpModel()
629 output_ = AddOutput(TensorType_FLOAT32); in SoftmaxOpModel()
678 output_ = AddOutput(TensorType_FLOAT32); in ReshapeOpModel()
711 output_ = AddOutput(output); in SqueezeOpModel()
768 output_ = AddOutput(output); in L2NormOpModel()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | detection_postprocess_test.cc | 50 output1_ = AddOutput(output1); in BaseDetectionPostprocessOpModel()
51 output2_ = AddOutput(output2); in BaseDetectionPostprocessOpModel()
52 output3_ = AddOutput(output3); in BaseDetectionPostprocessOpModel()
53 output4_ = AddOutput(output4); in BaseDetectionPostprocessOpModel()
259 output1_ = AddOutput(output1); in DetectionPostprocessOpModelwithRegularNMS()
260 output2_ = AddOutput(output2); in DetectionPostprocessOpModelwithRegularNMS()
261 output3_ = AddOutput(output3); in DetectionPostprocessOpModelwithRegularNMS()
262 output4_ = AddOutput(output4); in DetectionPostprocessOpModelwithRegularNMS()
|
| D | activations_test.cc | 34 output_ = AddOutput({input.type, {}, 0, 0, 1. / 256}); in BaseActivationsOpModel()
36 output_ = AddOutput({input.type, {}, 0, 0, 1. / 256, -128}); in BaseActivationsOpModel()
38 output_ = AddOutput({input.type, {}}); in BaseActivationsOpModel()
48 output_ = AddOutput({input.type, {}, 0, 0, 1. / 256}); in BaseActivationsOpModel()
50 output_ = AddOutput({TensorType_INT8, {}, 0, 0, 1. / 256, -128}); in BaseActivationsOpModel()
52 output_ = AddOutput({input.type, {}}); in BaseActivationsOpModel()
62 output_ = AddOutput(output); in BaseActivationsOpModel()
842 output_ = AddOutput({input.type, input.shape, input.min, input.max}); in BasePReluOpModel()
945 output_ = AddOutput(input); in LeakyReluOpModel()
|
| D | reshape_test.cc | 83 output_ = AddOutput(GetTensorType<T>()); in BuildWithHardcodedShape()
95 output_ = AddOutput(GetTensorType<T>()); in BuildWithTensorShape()
112 output_ = AddOutput(GetTensorType<T>()); in BuildWithConstantTensorShape()
|
| D | unique_test.cc | 32 output_id_ = AddOutput(input_type); in UniqueOpModel()
33 output_index_id_ = AddOutput(index_out_type); in UniqueOpModel()
|
| D | reduce_test.cc | 65 output_ = AddOutput(output); in MeanOpConstModel()
79 output_ = AddOutput(output); in MeanOpDynamicModel()
94 output_ = AddOutput(output); in SumOpConstModel()
108 output_ = AddOutput(output); in SumOpDynamicModel()
123 output_ = AddOutput(output); in ProdOpConstModel()
137 output_ = AddOutput(output); in ProdOpDynamicModel()
152 output_ = AddOutput(output); in MaxOpConstModel()
166 output_ = AddOutput(output); in MaxOpDynamicModel()
181 output_ = AddOutput(output); in MinOpConstModel()
195 output_ = AddOutput(output); in MinOpDynamicModel()
[all …]
|
| D | pad_test.cc | 82 this->output_ = this->AddOutput(output); in PadV2OpConstModel()
99 this->output_ = this->AddOutput(output); in PadV2OpConstModel()
122 output_ = AddOutput(output); in PadOpConstModel()
142 this->output_ = this->AddOutput(output); in PadV2OpDynamicModel()
155 this->output_ = this->AddOutput(output); in PadV2OpDynamicModel()
178 this->output_ = this->AddOutput(output); in PadOpDynamicModel()
|
| D | batch_to_space_nd_test.cc | 70 output_ = AddOutput(type); in BatchToSpaceNDOpConstModel()
94 output_ = AddOutput(type); in BatchToSpaceNDOpDynamicModel()
|
| D | elementwise_test.cc | 42 output_ = AddOutput(TensorType_FLOAT32); in ElementWiseOpFloatModel()
53 output_ = AddOutput(TensorType_BOOL); in ElementWiseOpBoolModel()
|
| D | hashtable_lookup_test.cc | 42 output_ = AddOutput(type); in HashtableLookupOpModel()
43 hit_ = AddOutput(TensorType_UINT8); in HashtableLookupOpModel()
|
| D | topk_v2_test.cc | 34 output_values_ = AddOutput(input_type); in TopKV2OpModel()
35 output_indexes_ = AddOutput(TensorType_INT32); in TopKV2OpModel()
|
| D | concatenation_test.cc | 39 output_ = AddOutput({input_template.type, /*shape=*/{}, input_template.min, in BaseConcatenationOpModel()
73 output_ = AddOutput({output_template.type, /*shape=*/{}, in QuantizedConcatenationOpModel()
|
| D | exp_test.cc | 30 output_ = AddOutput(output); in ExpOpModel()
|
| D | ceil_test.cc | 31 output_ = AddOutput(TensorType_FLOAT32); in CeilOpModel()
|
| D | floor_test.cc | 31 output_ = AddOutput(TensorType_FLOAT32); in FloorOpModel()
|
| D | dequantize_test.cc | 35 output_ = AddOutput({TensorType_FLOAT32, shape}); in DequantizeOpModel()
|
| /external/tensorflow/tensorflow/lite/models/smartreply/ops/ |
| D | extract_feature_test.cc | 39 signature_ = AddOutput(TensorType_INT32); in ExtractFeatureOpModel()
40 weight_ = AddOutput(TensorType_FLOAT32); in ExtractFeatureOpModel()
|
| D | predict_test.cc | 45 output_label_ = AddOutput(TensorType_INT32); in PredictOpModel()
46 output_weight_ = AddOutput(TensorType_FLOAT32); in PredictOpModel()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 114 node_map->AddOutput(NodeName(ctrl_input), node->name()); in MaybeAddControlInput()
279 node_map->AddOutput(node->name(), added_node->name()); in AddControlDependency()
399 node_map_->AddOutput(NodeName(ctrl_dep), node->name()); in MaterializeShapes()
488 node_map_->AddOutput(NodeName(ctrl_dep), added_node->name()); in MaterializeShapes()
491 node_map_->AddOutput(const_name, output->name()); in MaterializeShapes()
651 node_map_->AddOutput(NodeName(ctrl_dep), const_name); in MaterializeBroadcastGradientArgs()
765 node_map_->AddOutput(NodeName(ctrl_dep), const_name); in MaterializeReductionIndices()
1278 node_map_->AddOutput(node->name(), const_out->name()); in FoldMergeNode()
1291 node_map_->AddOutput(node->name(), const_index->name()); in FoldMergeNode()
1301 node_map_->AddOutput(const_out->name(), output->name()); in FoldMergeNode()
[all …]
|
| D | arithmetic_optimizer.cc | 125 node_map->AddOutput(NodeName(new_input), node->name()); in MaybeAddControlInput()
226 ctx().node_map->AddOutput(NodeName(src->input(i)), in ForwardControlDependencies()
627 ctx().node_map->AddOutput(inputAndShape.input, node_name); in AddInputsOfSymbolicallyEqualShape()
651 ctx().node_map->AddOutput(left.input, node_name); in AddAggregatedInputs()
652 ctx().node_map->AddOutput(right.input, node_name); in AddAggregatedInputs()
725 ctx().node_map->AddOutput(common_factor, new_outer_node->name()); in TrySimplify()
726 ctx().node_map->AddOutput(new_add_node->name(), new_outer_node->name()); in TrySimplify()
732 ctx().node_map->AddOutput(unique_factor_i, new_add_node->name()); in TrySimplify()
738 ctx().node_map->AddOutput(NodeName(ctrl_dep), new_add_node->name()); in TrySimplify()
1072 ctx().node_map->AddOutput(NodeName(input_0), node->name()); in UpdateInputs()
[all …]
|
| D | loop_optimizer.cc | 104 node_map_->AddOutput(NodeName(enter_input), consumer->name()); in HandleInvariantEnter()
110 node_map_->AddOutput(NodeName(control_input), consumer->name()); in HandleInvariantEnter()
153 node_map_->AddOutput(const_node->name(), consumer->name()); in HandleConst()
185 node_map_->AddOutput(NodeName(ctrl_dep), const_node->name()); in HandleConst()
251 node_map_->AddOutput(node->name(), new_enter->name()); in HandleInvariantNode()
252 node_map_->AddOutput(new_enter->name(), consumer->name()); in HandleInvariantNode()
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | ctc_beam_search_decoder_test.cc | 51 outputs_.push_back(AddOutput(TensorType_INT32)); in CTCBeamSearchDecoderOpModel()
53 outputs_.push_back(AddOutput(TensorType_FLOAT32)); in CTCBeamSearchDecoderOpModel()
|