Home
last modified time | relevance | path

Searched refs:AddInput (Results 1 – 25 of 265) sorted by relevance

1234567891011

/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dconv_2d_builder.cc254 AddInput(graph_builder_->GetHexagonTensorId(inputs->data[0])); in PopulateSubGraph()
255 AddInput(TensorID(dilation_factors_const->GetID(), 0)); in PopulateSubGraph()
256 AddInput(TensorID(paddings_const->GetID(), 0)); in PopulateSubGraph()
257 AddInput(TensorID(data_min_const->GetID(), 0)); in PopulateSubGraph()
258 AddInput(TensorID(data_max_const->GetID(), 0)); in PopulateSubGraph()
272 conv_op->AddInput(space_to_batch_op_out); in PopulateSubGraph()
273 conv_op->AddInput(graph_builder_->GetHexagonTensorId(inputs->data[1])); in PopulateSubGraph()
274 conv_op->AddInput(TensorID(data_min_const->GetID(), 0)); in PopulateSubGraph()
275 conv_op->AddInput(TensorID(data_max_const->GetID(), 0)); in PopulateSubGraph()
276 conv_op->AddInput(TensorID(weights_min_node_->GetID(), 0)); in PopulateSubGraph()
[all …]
Darithmetic_builder.cc35 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
40 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
60 AddInput(TensorID(output_min_const->GetID(), 0)); in PopulateSubGraph()
61 AddInput(TensorID(output_max_const->GetID(), 0)); in PopulateSubGraph()
73 requantize_op->AddInput(math_out); in PopulateSubGraph()
74 requantize_op->AddInput(math_out_min); in PopulateSubGraph()
75 requantize_op->AddInput(math_out_max); in PopulateSubGraph()
76 requantize_op->AddInput(TensorID(output_min_const->GetID(), 0)); in PopulateSubGraph()
77 requantize_op->AddInput(TensorID(output_max_const->GetID(), 0)); in PopulateSubGraph()
93 requantize_op->AddInput(result_out); in PopulateSubGraph()
[all …]
Drsqrt_builder.cc98 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
99 AddInput(TensorID(min_const->GetID(), 0)); in PopulateSubGraph()
100 AddInput(TensorID(max_const->GetID(), 0)); in PopulateSubGraph()
114 reshape_op->AddInput(sqrt_output); in PopulateSubGraph()
115 reshape_op->AddInput(TensorID(target_shape_node->GetID(), 0)); in PopulateSubGraph()
138 div_op->AddInput(numerator_out_); in PopulateSubGraph()
139 div_op->AddInput(reshape_out); in PopulateSubGraph()
140 div_op->AddInput(numerator_min_); in PopulateSubGraph()
141 div_op->AddInput(numerator_max_); in PopulateSubGraph()
142 div_op->AddInput(sqrt_output_min); in PopulateSubGraph()
[all …]
Dmatmul_builder.cc65 matmul_op->AddInput(graph_builder->GetHexagonTensorId(data_tensor_id)); in AddFullyConnectedHelper()
66 matmul_op->AddInput(weights_id); in AddFullyConnectedHelper()
67 matmul_op->AddInput(OpBuilder::TensorID(data_min_const->GetID(), 0)); in AddFullyConnectedHelper()
68 matmul_op->AddInput(OpBuilder::TensorID(data_max_const->GetID(), 0)); in AddFullyConnectedHelper()
69 matmul_op->AddInput(weights_min_id); in AddFullyConnectedHelper()
70 matmul_op->AddInput(weights_max_id); in AddFullyConnectedHelper()
104 bias_add_op->AddInput(matmul_out); in AddFullyConnectedHelper()
105 bias_add_op->AddInput(graph_builder->GetHexagonTensorId(bias_tensor_id)); in AddFullyConnectedHelper()
106 bias_add_op->AddInput(matmul_out_min); in AddFullyConnectedHelper()
107 bias_add_op->AddInput(matmul_out_max); in AddFullyConnectedHelper()
[all …]
Dsquared_difference.cc46 AddInput(graph_builder_->GetHexagonTensorId(tensor_a_index)); in PopulateSubGraph()
47 AddInput(graph_builder_->GetHexagonTensorId(tensor_b_index)); in PopulateSubGraph()
73 mul_op->AddInput(sub_out); in PopulateSubGraph()
74 mul_op->AddInput(sub_out); in PopulateSubGraph()
75 mul_op->AddInput(sub_min); in PopulateSubGraph()
76 mul_op->AddInput(sub_max); in PopulateSubGraph()
77 mul_op->AddInput(sub_min); in PopulateSubGraph()
78 mul_op->AddInput(sub_max); in PopulateSubGraph()
79 mul_op->AddInput(TensorID(output_min_const->GetID(), 0)); in PopulateSubGraph()
80 mul_op->AddInput(TensorID(output_max_const->GetID(), 0)); in PopulateSubGraph()
Dreduce_builder.cc35 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
46 AddInput(TensorID(const_axes_node->GetID(), 0)); in PopulateSubGraph()
65 AddInput(TensorID(output_min_const->GetID(), 0)); in PopulateSubGraph()
66 AddInput(TensorID(output_max_const->GetID(), 0)); in PopulateSubGraph()
81 requantize_op->AddInput(mean_output); in PopulateSubGraph()
82 requantize_op->AddInput(mean_out_min); in PopulateSubGraph()
83 requantize_op->AddInput(mean_out_max); in PopulateSubGraph()
84 requantize_op->AddInput(TensorID(output_min_const->GetID(), 0)); in PopulateSubGraph()
85 requantize_op->AddInput(TensorID(output_max_const->GetID(), 0)); in PopulateSubGraph()
Dpack_builder.cc49 AddInput(TensorID(axis_node->GetID(), 0)); in PopulateSubGraph()
59 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
70 AddInput(TensorID(data_min_const->GetID(), 0)); in PopulateSubGraph()
77 AddInput(TensorID(data_max_const->GetID(), 0)); in PopulateSubGraph()
105 requantize_op->AddInput(pack_out); in PopulateSubGraph()
106 requantize_op->AddInput(pack_out_min); in PopulateSubGraph()
107 requantize_op->AddInput(pack_out_max); in PopulateSubGraph()
108 requantize_op->AddInput(TensorID(output_min_const->GetID(), 0)); in PopulateSubGraph()
109 requantize_op->AddInput(TensorID(output_max_const->GetID(), 0)); in PopulateSubGraph()
Dconcat_builder.cc40 AddInput(TensorID(axis_const->GetID(), 0)); in PopulateSubGraph()
55 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
69 AddInput(TensorID(data_min_const->GetID(), 0)); in PopulateSubGraph()
77 AddInput(TensorID(data_max_const->GetID(), 0)); in PopulateSubGraph()
112 requantize_op->AddInput(concat_out); in PopulateSubGraph()
113 requantize_op->AddInput(concat_out_min); in PopulateSubGraph()
114 requantize_op->AddInput(concat_out_max); in PopulateSubGraph()
115 requantize_op->AddInput(TensorID(output_min_const->GetID(), 0)); in PopulateSubGraph()
116 requantize_op->AddInput(TensorID(output_max_const->GetID(), 0)); in PopulateSubGraph()
Dpool_2d_builder.cc35 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
53 AddInput(TensorID(filter_node->GetID(), 0)); in PopulateSubGraph()
58 AddInput(TensorID(stride_node->GetID(), 0)); in PopulateSubGraph()
91 requantize_op->AddInput(pool_out); in PopulateSubGraph()
92 requantize_op->AddInput(pool_out_min); in PopulateSubGraph()
93 requantize_op->AddInput(pool_out_max); in PopulateSubGraph()
94 requantize_op->AddInput(TensorID(output_min_const->GetID(), 0)); in PopulateSubGraph()
95 requantize_op->AddInput(TensorID(output_max_const->GetID(), 0)); in PopulateSubGraph()
Dstrided_slice_builder.cc32 AddInput(graph_builder_->GetHexagonTensorId(inputs->data[0])); in PopulateSubGraph()
43 AddInput(TensorID(begins_node->GetID(), 0)); in PopulateSubGraph()
44 AddInput(TensorID(ends_node->GetID(), 0)); in PopulateSubGraph()
45 AddInput(TensorID(steps_node->GetID(), 0)); in PopulateSubGraph()
60 AddInput(TensorID(begin_mask_const->GetID(), 0)); in PopulateSubGraph()
63 AddInput(TensorID(end_mask_const->GetID(), 0)); in PopulateSubGraph()
67 AddInput(TensorID(shrink_axis_mask_const->GetID(), 0)); in PopulateSubGraph()
Dtranspose_conv_2d_builder.cc45 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
67 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
87 AddInput(TensorID(weights_min_const->GetID(), 0)); in PopulateSubGraph()
88 AddInput(TensorID(weights_max_const->GetID(), 0)); in PopulateSubGraph()
112 AddInput(TensorID(padding_const->GetID(), 0)); in PopulateSubGraph()
121 AddInput(TensorID(stride_node->GetID(), 0)); in PopulateSubGraph()
166 AddInput(TensorID(bias_const->GetID(), 0)); in PopulateSubGraph()
167 AddInput(TensorID(bias_min_const->GetID(), 0)); in PopulateSubGraph()
168 AddInput(TensorID(bias_max_const->GetID(), 0)); in PopulateSubGraph()
176 AddInput(TensorID(per_channel_quant_.channel_scales_node->GetID(), 0)); in PopulateSubGraph()
Dresize_nearest_neighbor_builder.cc34 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
44 AddInput(TensorID(const_output_dim_node->GetID(), 0)); in PopulateSubGraph()
46 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
59 AddInput(TensorID(align_corners_const->GetID(), 0)); in PopulateSubGraph()
64 AddInput(TensorID(half_pixel_centers_const->GetID(), 0)); in PopulateSubGraph()
/external/tensorflow/tensorflow/lite/kernels/
Doptional_tensor_test.cc39 input_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
44 input_to_input_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
47 input_to_forget_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
48 input_to_cell_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
49 input_to_output_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
54 recurrent_to_input_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
57 recurrent_to_forget_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
58 recurrent_to_cell_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
59 recurrent_to_output_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
65 cell_to_input_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
[all …]
Dlstm_test.cc48 input_ = AddInput({TensorType_FLOAT32, {n_batch, n_input}}); in LSTMOpModel()
53 input_to_input_weights_ = AddInput({weight_type, {n_cell, n_input}}); in LSTMOpModel()
55 input_to_forget_weights_ = AddInput({weight_type, {n_cell, n_input}}); in LSTMOpModel()
56 input_to_cell_weights_ = AddInput({weight_type, {n_cell, n_input}}); in LSTMOpModel()
57 input_to_output_weights_ = AddInput({weight_type, {n_cell, n_input}}); in LSTMOpModel()
62 recurrent_to_input_weights_ = AddInput({weight_type, {n_cell, n_output}}); in LSTMOpModel()
64 recurrent_to_forget_weights_ = AddInput({weight_type, {n_cell, n_output}}); in LSTMOpModel()
65 recurrent_to_cell_weights_ = AddInput({weight_type, {n_cell, n_output}}); in LSTMOpModel()
66 recurrent_to_output_weights_ = AddInput({weight_type, {n_cell, n_output}}); in LSTMOpModel()
72 cell_to_input_weights_ = AddInput({weight_type, {n_cell}}); in LSTMOpModel()
[all …]
Dbidirectional_sequence_lstm_test.cc48 input_ = AddInput(TensorType_FLOAT32); in BidirectionalLSTMOpModel()
55 fw_input_to_input_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
58 fw_input_to_forget_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
59 fw_input_to_cell_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
60 fw_input_to_output_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
65 fw_recurrent_to_input_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
68 fw_recurrent_to_forget_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
69 fw_recurrent_to_cell_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
70 fw_recurrent_to_output_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
76 fw_cell_to_input_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
[all …]
Dnon_max_suppression_test.cc74 input_boxes_ = AddInput({TensorType_FLOAT32, {num_boxes, 4}}); in NonMaxSuppressionV4OpModel()
75 input_scores_ = AddInput({TensorType_FLOAT32, {num_boxes}}); in NonMaxSuppressionV4OpModel()
80 input_max_output_size_ = AddInput(TensorType_INT32); in NonMaxSuppressionV4OpModel()
83 input_score_threshold_ = AddInput({TensorType_FLOAT32, {}}); in NonMaxSuppressionV4OpModel()
165 input_boxes_ = AddInput({TensorType_FLOAT32, {num_boxes, 4}}); in NonMaxSuppressionV5OpModel()
166 input_scores_ = AddInput({TensorType_FLOAT32, {num_boxes}}); in NonMaxSuppressionV5OpModel()
171 input_max_output_size_ = AddInput(TensorType_INT32); in NonMaxSuppressionV5OpModel()
174 input_score_threshold_ = AddInput({TensorType_FLOAT32, {}}); in NonMaxSuppressionV5OpModel()
Dunidirectional_sequence_lstm_test.cc46 input_ = AddInput(TensorType_FLOAT32); in UnidirectionalLSTMOpModel()
51 input_to_input_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
54 input_to_forget_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
55 input_to_cell_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
56 input_to_output_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
61 recurrent_to_input_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
64 recurrent_to_forget_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
65 recurrent_to_cell_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
66 recurrent_to_output_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
72 cell_to_input_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
[all …]
/external/tensorflow/tensorflow/c/experimental/ops/
Dmath_ops.cc34 TF_RETURN_IF_ERROR(mul_op->AddInput(inputs[0])); in Mul()
35 TF_RETURN_IF_ERROR(mul_op->AddInput(inputs[1])); in Mul()
52 TF_RETURN_IF_ERROR(conj_op->AddInput(inputs[0])); in Conj()
68 TF_RETURN_IF_ERROR(add_op->AddInput(inputs[0])); in Add()
69 TF_RETURN_IF_ERROR(add_op->AddInput(inputs[1])); in Add()
81 TF_RETURN_IF_ERROR(sub_op->AddInput(inputs[0])); in Sub()
82 TF_RETURN_IF_ERROR(sub_op->AddInput(inputs[1])); in Sub()
96 TF_RETURN_IF_ERROR(matmul_op->AddInput(inputs[0])); in MatMul()
97 TF_RETURN_IF_ERROR(matmul_op->AddInput(inputs[1])); in MatMul()
112 TF_RETURN_IF_ERROR(neg_op->AddInput(inputs[0])); in Neg()
[all …]
Dnn_ops.cc34 TF_RETURN_IF_ERROR(sm_loss_op->AddInput(inputs[0])); // input scores in SparseSoftmaxCrossEntropyWithLogits()
35 TF_RETURN_IF_ERROR(sm_loss_op->AddInput(inputs[1])); // labels in SparseSoftmaxCrossEntropyWithLogits()
51 TF_RETURN_IF_ERROR(relugrad_op->AddInput(inputs[0])); // upstream grads in ReluGrad()
52 TF_RETURN_IF_ERROR(relugrad_op->AddInput(inputs[1])); // relu inputs in ReluGrad()
65 TF_RETURN_IF_ERROR(relu_op->AddInput(inputs[0])); in Relu()
79 TF_RETURN_IF_ERROR(bias_add_op->AddInput(inputs[0])); // tensor input in BiasAdd()
80 TF_RETURN_IF_ERROR(bias_add_op->AddInput(inputs[1])); // bias in BiasAdd()
98 TF_RETURN_IF_ERROR(bias_add_grad_op->AddInput(inputs[0])); in BiasAddGrad()
Darray_ops.cc32 TF_RETURN_IF_ERROR(identity_op->AddInput(inputs[0])); in Identity()
55 TF_RETURN_IF_ERROR(z_op->AddInput(inputs[0])); in ZerosLike()
66 TF_RETURN_IF_ERROR(shape_op->AddInput(inputs[0])); // input in Shape()
78 TF_RETURN_IF_ERROR(op->AddInput(inputs[0])); in ExpandDims()
79 TF_RETURN_IF_ERROR(op->AddInput(inputs[1])); in ExpandDims()
90 TF_RETURN_IF_ERROR(op->AddInput(inputs[0])); in OnesLike()
/external/tensorflow/tensorflow/core/kernels/
Dsave_op_test.cc70 AddInput<tstring>(TensorShape({}), in TEST_F()
74 AddInput<tstring>(TensorShape({14}), [&tensornames](int x) -> tstring { in TEST_F()
79 AddInput<bool>(TensorShape({2}), [](int x) -> bool { return x != 0; }); in TEST_F()
82 AddInput<int32>(TensorShape({10}), [](int x) -> int32 { return x + 1; }); in TEST_F()
85 AddInput<float>(TensorShape({2, 4}), in TEST_F()
89 AddInput<double>(TensorShape({2, 4}), in TEST_F()
93 AddInput<qint8>(TensorShape({3, 2}), in TEST_F()
97 AddInput<qint32>(TensorShape({2, 3}), [](int x) -> qint32 { in TEST_F()
102 AddInput<uint8>(TensorShape({11}), [](int x) -> uint8 { return x + 1; }); in TEST_F()
105 AddInput<int8>(TensorShape({7}), [](int x) -> int8 { return x - 7; }); in TEST_F()
[all …]
Dsave_v2_op_test.cc63 AddInput<tstring>(TensorShape({}), in TEST_F()
67 AddInput<tstring>(TensorShape({13}), [&tensornames](int x) -> tstring { in TEST_F()
72 AddInput<tstring>(TensorShape({13}), in TEST_F()
76 AddInput<bool>(TensorShape({2}), [](int x) -> bool { return x != 0; }); in TEST_F()
79 AddInput<int32>(TensorShape({10}), [](int x) -> int32 { return x + 1; }); in TEST_F()
82 AddInput<float>(TensorShape({2, 4}), in TEST_F()
86 AddInput<double>(TensorShape({2, 4}), in TEST_F()
90 AddInput<qint8>(TensorShape({3, 2}), in TEST_F()
94 AddInput<qint32>(TensorShape({2, 3}), [](int x) -> qint32 { in TEST_F()
99 AddInput<uint8>(TensorShape({11}), [](int x) -> uint8 { return x + 1; }); in TEST_F()
[all …]
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dadd_op_builder.cc56 activation_builder->AddInput(AddOutput()); in PopulateSubgraph()
74 AddInput(inputs->data[1]); in RegisterInputs()
77 AddInput(inputs->data[0]); in RegisterInputs()
80 AddInput(inputs->data[0]); in RegisterInputs()
81 AddInput(inputs->data[1]); in RegisterInputs()
Dmul_op_builder.cc62 activation_builder->AddInput(AddOutput()); in PopulateSubgraph()
80 AddInput(inputs->data[1]); in RegisterInputs()
83 AddInput(inputs->data[0]); in RegisterInputs()
86 AddInput(inputs->data[0]); in RegisterInputs()
87 AddInput(inputs->data[1]); in RegisterInputs()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc185 input1_ = AddInput(input1); in Init()
186 input2_ = AddInput(input2); in Init()
493 input1_ = AddInput(input1); in FloatMulOpModel()
494 input2_ = AddInput(input2); in FloatMulOpModel()
527 input_ = AddInput(input); in FloatPoolingOpModel()
598 input_ = AddInput(input); in ConvolutionOpModel()
599 filter_ = AddInput(filter); in ConvolutionOpModel()
603 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); in ConvolutionOpModel()
610 bias_ = AddInput(bias); in ConvolutionOpModel()
961 input_ = AddInput(input); in PerChannelQuantizedConvolutionWithConstantFilterOpModel()
[all …]

1234567891011