/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
D | conv_2d_builder.cc | 254 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 …]
|
D | arithmetic_builder.cc | 35 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 …]
|
D | rsqrt_builder.cc | 98 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 …]
|
D | matmul_builder.cc | 65 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 …]
|
D | squared_difference.cc | 46 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()
|
D | reduce_builder.cc | 35 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()
|
D | pack_builder.cc | 49 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()
|
D | concat_builder.cc | 40 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()
|
D | pool_2d_builder.cc | 35 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()
|
D | strided_slice_builder.cc | 32 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()
|
D | transpose_conv_2d_builder.cc | 45 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()
|
D | resize_nearest_neighbor_builder.cc | 34 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/ |
D | optional_tensor_test.cc | 39 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 …]
|
D | lstm_test.cc | 48 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 …]
|
D | bidirectional_sequence_lstm_test.cc | 48 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 …]
|
D | non_max_suppression_test.cc | 74 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()
|
D | unidirectional_sequence_lstm_test.cc | 46 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/ |
D | math_ops.cc | 34 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 …]
|
D | nn_ops.cc | 34 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()
|
D | array_ops.cc | 32 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/ |
D | save_op_test.cc | 70 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 …]
|
D | save_v2_op_test.cc | 63 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/ |
D | add_op_builder.cc | 56 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()
|
D | mul_op_builder.cc | 62 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/ |
D | nnapi_delegate_test.cc | 185 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 …]
|