Home
last modified time | relevance | path

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

12345

/external/tensorflow/tensorflow/lite/kernels/
Doptional_tensor_test.cc41 input_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
46 input_to_input_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
49 input_to_forget_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
50 input_to_cell_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
51 input_to_output_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
56 recurrent_to_input_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
59 recurrent_to_forget_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
60 recurrent_to_cell_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
61 recurrent_to_output_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
67 cell_to_input_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
[all …]
Dbidirectional_sequence_lstm_test.cc52 input_ = AddInput(TensorType_FLOAT32); in BidirectionalLSTMOpModel()
59 fw_input_to_input_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
62 fw_input_to_forget_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
63 fw_input_to_cell_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
64 fw_input_to_output_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
69 fw_recurrent_to_input_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
72 fw_recurrent_to_forget_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
73 fw_recurrent_to_cell_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
74 fw_recurrent_to_output_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
80 fw_cell_to_input_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
[all …]
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 …]
Dembedding_lookup_sparse_test.cc39 lookup_ = AddInput(TensorType_INT32); in EmbeddingLookupSparseOpModel()
40 indices_ = AddInput(TensorType_INT32); in EmbeddingLookupSparseOpModel()
41 dense_shape_ = AddInput(TensorType_INT32); in EmbeddingLookupSparseOpModel()
42 weights_ = AddInput(TensorType_FLOAT32); in EmbeddingLookupSparseOpModel()
43 value_ = AddInput(TensorType_FLOAT32); in EmbeddingLookupSparseOpModel()
Dlstm_test.cc47 input_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
52 input_to_input_weights_ = AddInput(weight_type); in LSTMOpModel()
55 input_to_forget_weights_ = AddInput(weight_type); in LSTMOpModel()
56 input_to_cell_weights_ = AddInput(weight_type); in LSTMOpModel()
57 input_to_output_weights_ = AddInput(weight_type); in LSTMOpModel()
62 recurrent_to_input_weights_ = AddInput(weight_type); in LSTMOpModel()
65 recurrent_to_forget_weights_ = AddInput(weight_type); in LSTMOpModel()
66 recurrent_to_cell_weights_ = AddInput(weight_type); in LSTMOpModel()
67 recurrent_to_output_weights_ = AddInput(weight_type); in LSTMOpModel()
73 cell_to_input_weights_ = AddInput(weight_type); in LSTMOpModel()
[all …]
Dbatch_to_space_nd_test.cc67 input_ = AddInput(type); in BatchToSpaceNDOpConstModel()
91 input_ = AddInput(type); in BatchToSpaceNDOpDynamicModel()
92 block_shape_ = AddInput(TensorType_INT32); in BatchToSpaceNDOpDynamicModel()
93 crops_ = AddInput(TensorType_INT32); in BatchToSpaceNDOpDynamicModel()
Dbasic_rnn_test.cc180 input_ = AddInput(TensorType_FLOAT32); in RNNOpModel()
181 weights_ = AddInput(weights); in RNNOpModel()
182 recurrent_weights_ = AddInput(recurrent_weights); in RNNOpModel()
183 bias_ = AddInput(TensorType_FLOAT32); in RNNOpModel()
184 hidden_state_ = AddInput(TensorType_FLOAT32, true); in RNNOpModel()
Dpad_test.cc76 this->input_ = this->AddInput(input); in PadV2OpConstModel()
94 this->input_ = this->AddInput(input); in PadV2OpConstModel()
97 this->constant_values_ = this->AddInput(constant_values); in PadV2OpConstModel()
119 this->input_ = AddInput(input); in PadOpConstModel()
138 this->input_ = this->AddInput(input); in PadV2OpDynamicModel()
139 this->paddings_ = this->AddInput(TensorType_INT32); in PadV2OpDynamicModel()
152 this->input_ = this->AddInput(input); in PadV2OpDynamicModel()
153 this->paddings_ = this->AddInput(TensorType_INT32); in PadV2OpDynamicModel()
154 this->constant_values_ = this->AddInput(constant_values); in PadV2OpDynamicModel()
175 this->input_ = this->AddInput(input); in PadOpDynamicModel()
[all …]
Dsparse_to_dense_test.cc36 indices_ = AddInput(tensor_index_type); in SparseToDenseOpModel()
37 output_shape_ = AddInput(TensorType_INT32); in SparseToDenseOpModel()
38 values_ = AddInput(tensor_input_type); in SparseToDenseOpModel()
39 default_value_ = AddInput(tensor_input_type); in SparseToDenseOpModel()
Done_hot_test.cc35 indices_ = AddInput(indices_type); in OneHotOpModel()
36 int depth = AddInput(TensorType_INT32); in OneHotOpModel()
37 int on = AddInput(dtype); in OneHotOpModel()
38 int off = AddInput(dtype); in OneHotOpModel()
Dlsh_projection_test.cc35 hash_ = AddInput(TensorType_FLOAT32); in LSHProjectionOpModel()
36 input_ = AddInput(TensorType_INT32); in LSHProjectionOpModel()
38 weight_ = AddInput(TensorType_FLOAT32); in LSHProjectionOpModel()
Dbidirectional_sequence_rnn_test.cc664 input_ = AddInput(TensorType_FLOAT32); in BidirectionalRNNOpModel()
665 fw_weights_ = AddInput(TensorType_FLOAT32); in BidirectionalRNNOpModel()
666 fw_recurrent_weights_ = AddInput(TensorType_FLOAT32); in BidirectionalRNNOpModel()
667 fw_bias_ = AddInput(TensorType_FLOAT32); in BidirectionalRNNOpModel()
668 fw_hidden_state_ = AddInput(TensorType_FLOAT32, true); in BidirectionalRNNOpModel()
669 bw_weights_ = AddInput(TensorType_FLOAT32); in BidirectionalRNNOpModel()
670 bw_recurrent_weights_ = AddInput(TensorType_FLOAT32); in BidirectionalRNNOpModel()
671 bw_bias_ = AddInput(TensorType_FLOAT32); in BidirectionalRNNOpModel()
672 bw_hidden_state_ = AddInput(TensorType_FLOAT32, true); in BidirectionalRNNOpModel()
676 aux_input_ = AddInput(TensorType_FLOAT32); in BidirectionalRNNOpModel()
Dsvdf_test.cc140 input_ = AddInput(TensorType_FLOAT32); in BaseSVDFOpModel()
141 weights_feature_ = AddInput(weights_feature_type); in BaseSVDFOpModel()
142 weights_time_ = AddInput(weights_time_type); in BaseSVDFOpModel()
145 activation_state_ = AddInput( in BaseSVDFOpModel()
Dreshape_test.cc82 input_ = AddInput({GetTensorType<T>(), input_shape}); in BuildWithHardcodedShape()
94 input_ = AddInput({GetTensorType<T>(), input_shape}); in BuildWithTensorShape()
96 int shape_input_tensor = AddInput({TensorType_INT32, shape_shape}); in BuildWithTensorShape()
111 input_ = AddInput({GetTensorType<T>(), input_shape}); in BuildWithConstantTensorShape()
Dreduce_test.cc63 input_ = AddInput(input); in MeanOpConstModel()
77 input_ = AddInput(input); in MeanOpDynamicModel()
78 axis_ = AddInput(axis); in MeanOpDynamicModel()
92 input_ = AddInput(input); in SumOpConstModel()
106 input_ = AddInput(input); in SumOpDynamicModel()
107 axis_ = AddInput(axis); in SumOpDynamicModel()
121 input_ = AddInput(input); in ProdOpConstModel()
135 input_ = AddInput(input); in ProdOpDynamicModel()
136 axis_ = AddInput(axis); in ProdOpDynamicModel()
150 input_ = AddInput(input); in MaxOpConstModel()
[all …]
Ddepthwise_conv_test.cc49 input_ = AddInput(input); in BaseDepthwiseConvolutionOpModel()
50 filter_ = AddInput(filter); in BaseDepthwiseConvolutionOpModel()
54 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); in BaseDepthwiseConvolutionOpModel()
81 bias_ = AddInput(bias); in BaseDepthwiseConvolutionOpModel()
86 bias_ = AddInput(bias); in BaseDepthwiseConvolutionOpModel()
Drange_test.cc30 start_ = AddInput(dtype); in RangeOpModel()
31 limit_ = AddInput(dtype); in RangeOpModel()
32 delta_ = AddInput(dtype); in RangeOpModel()
/external/tensorflow/tensorflow/core/kernels/
Dsave_op_test.cc68 AddInput<string>(TensorShape({}), in TEST_F()
72 AddInput<string>(TensorShape({14}), 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()
102 AddInput<int8>(TensorShape({7}), [](int x) -> int8 { return x - 7; }); in TEST_F()
[all …]
Dsave_v2_op_test.cc62 AddInput<string>(TensorShape({}), in TEST_F()
66 AddInput<string>(TensorShape({13}), in TEST_F()
70 AddInput<string>(TensorShape({13}), in TEST_F()
74 AddInput<bool>(TensorShape({2}), [](int x) -> bool { return x != 0; }); in TEST_F()
77 AddInput<int32>(TensorShape({10}), [](int x) -> int32 { return x + 1; }); in TEST_F()
80 AddInput<float>(TensorShape({2, 4}), in TEST_F()
84 AddInput<double>(TensorShape({2, 4}), in TEST_F()
88 AddInput<qint8>(TensorShape({3, 2}), in TEST_F()
92 AddInput<qint32>(TensorShape({2, 3}), [](int x) -> qint32 { in TEST_F()
97 AddInput<uint8>(TensorShape({11}), [](int x) -> uint8 { return x + 1; }); in TEST_F()
[all …]
Dops_testutil.h116 void AddInput(const TensorShape& shape, std::function<T(int)> input_mapping) { in AddInput() function
117 test::FillFn(AddInput(DataTypeToEnum<T>::v(), shape), input_mapping); in AddInput()
124 test::FillValues<T>(AddInput(DataTypeToEnum<T>::v(), shape), data); in AddInputFromArray()
132 test::FillValues<T>(AddInput(DataTypeToEnum<T>::v(), shape), data); in AddInputFromList()
208 Tensor* AddInput(DataType dtype, const TensorShape& shape) { in AddInput() function
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc95 input1_ = AddInput(input1); in FloatAddOpModel()
96 input2_ = AddInput(input2); in FloatAddOpModel()
166 input1_ = AddInput(input1); in FloatMulOpModel()
167 input2_ = AddInput(input2); in FloatMulOpModel()
201 input_ = AddInput(input); in FloatPoolingOpModel()
272 input_ = AddInput(input); in ConvolutionOpModel()
273 filter_ = AddInput(filter); in ConvolutionOpModel()
277 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); in ConvolutionOpModel()
284 bias_ = AddInput(bias); in ConvolutionOpModel()
451 input_ = AddInput(input); in DepthwiseConvolutionOpModel()
[all …]
/external/libtextclassifier/utils/tflite/
Dtoken_encoder_test.cc74 input_num_tokens_ = AddInput(tflite::TensorType_INT32); in TokenEncoderOpModel()
75 input_length_ = AddInput(tflite::TensorType_INT32); in TokenEncoderOpModel()
76 input_output_maxlength_ = AddInput(tflite::TensorType_INT32); in TokenEncoderOpModel()
77 input_attributes_int32_ = AddInput(tflite::TensorType_INT32); in TokenEncoderOpModel()
78 input_attributes_float_ = AddInput(tflite::TensorType_FLOAT32); in TokenEncoderOpModel()
Dtext_encoder_test.cc87 input_string_ = AddInput(tflite::TensorType_STRING); in TextEncoderOpModel()
88 input_length_ = AddInput(tflite::TensorType_INT32); in TextEncoderOpModel()
89 input_output_maxlength_ = AddInput(tflite::TensorType_INT32); in TextEncoderOpModel()
90 input_attributes_int32_ = AddInput(tflite::TensorType_INT32); in TextEncoderOpModel()
91 input_attributes_float_ = AddInput(tflite::TensorType_FLOAT32); in TextEncoderOpModel()
Ddist_diversification_test.cc56 distance_matrix_ = AddInput(tflite::TensorType_FLOAT32); in DistanceDiversificationOpModel()
57 min_distance_ = AddInput(tflite::TensorType_FLOAT32); in DistanceDiversificationOpModel()
58 num_results_ = AddInput(tflite::TensorType_INT32); in DistanceDiversificationOpModel()
/external/tensorflow/tensorflow/lite/models/smartreply/ops/
Dpredict_test.cc41 input_signature_ = AddInput(TensorType_INT32); in PredictOpModel()
42 model_key_ = AddInput(TensorType_INT32); in PredictOpModel()
43 model_label_ = AddInput(TensorType_INT32); in PredictOpModel()
44 model_weight_ = AddInput(TensorType_FLOAT32); in PredictOpModel()

12345