| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| optional_tensor_test.cc | 113 PopulateTensor(input_to_input_weights_, f);
117 PopulateTensor(input_to_forget_weights_, f);
121 PopulateTensor(input_to_cell_weights_, f);
125 PopulateTensor(input_to_output_weights_, f);
129 PopulateTensor(recurrent_to_input_weights_, f);
133 PopulateTensor(recurrent_to_forget_weights_, f);
137 PopulateTensor(recurrent_to_cell_weights_, f);
141 PopulateTensor(recurrent_to_output_weights_, f);
145 PopulateTensor(cell_to_input_weights_, f);
149 PopulateTensor(cell_to_forget_weights_, f)
[all...] |
| basic_rnn_test.cc | 144 void SetBias(std::initializer_list<float> f) { PopulateTensor(bias_, f); }
147 PopulateTensor(weights_, f);
151 PopulateTensor(recurrent_weights_, f);
155 PopulateTensor(input_, data);
159 PopulateTensor(input_, offset, begin, end);
166 PopulateTensor(hidden_state_, 0, zero_buffer.get(),
|
| mul_test.cc | 76 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 0.7, 0.8});
77 m.PopulateTensor<float>(m.input2(), {0.1, 0.2, 0.3, 0.5});
87 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 0.7, 0.8});
88 m.PopulateTensor<float>(m.input2(), {0.1, 0.2, 0.3, 5});
101 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 0.7, 0.8, 1.1, 2.0});
102 m.PopulateTensor<float>(m.input2(), {0.1, 0.2, 0.3, 0.5, 1.1, 0.1});
118 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 0.7, 0.8, 1.1, 2.0});
119 m.PopulateTensor<float>(m.input2(), {0.1});
|
| lstm_test.cc | 115 PopulateTensor(input_to_input_weights_, f);
119 PopulateTensor(input_to_forget_weights_, f);
123 PopulateTensor(input_to_cell_weights_, f);
127 PopulateTensor(input_to_output_weights_, f);
131 PopulateTensor(recurrent_to_input_weights_, f);
135 PopulateTensor(recurrent_to_forget_weights_, f);
139 PopulateTensor(recurrent_to_cell_weights_, f);
143 PopulateTensor(recurrent_to_output_weights_, f);
147 PopulateTensor(cell_to_input_weights_, f);
151 PopulateTensor(cell_to_forget_weights_, f)
[all...] |
| unidirectional_sequence_lstm_test.cc | 119 PopulateTensor(input_to_input_weights_, f);
123 PopulateTensor(input_to_forget_weights_, f);
127 PopulateTensor(input_to_cell_weights_, f);
131 PopulateTensor(input_to_output_weights_, f);
135 PopulateTensor(recurrent_to_input_weights_, f);
139 PopulateTensor(recurrent_to_forget_weights_, f);
143 PopulateTensor(recurrent_to_cell_weights_, f);
147 PopulateTensor(recurrent_to_output_weights_, f);
151 PopulateTensor(cell_to_input_weights_, f);
155 PopulateTensor(cell_to_forget_weights_, f)
[all...] |
| unidirectional_sequence_rnn_test.cc | 157 void SetBias(std::initializer_list<float> f) { PopulateTensor(bias_, f); }
160 PopulateTensor(weights_, f);
164 PopulateTensor(recurrent_weights_, f);
168 PopulateTensor(input_, data);
172 PopulateTensor(input_, offset, begin, end);
179 PopulateTensor(hidden_state_, 0, zero_buffer.get(),
|
| lsh_projection_test.cc | 60 PopulateTensor(input_, data);
64 PopulateTensor(hash_, data);
67 void SetWeight(std::initializer_list<float> f) { PopulateTensor(weight_, f); }
|
| add_test.cc | 76 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 0.7, 0.8});
77 m.PopulateTensor<float>(m.input2(), {0.1, 0.2, 0.3, 0.5});
86 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 0.7, 0.8});
87 m.PopulateTensor<float>(m.input2(), {0.1, 0.2, 0.3, 0.5});
99 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 0.7, 0.8, 1.1, 2.0});
100 m.PopulateTensor<float>(m.input2(), {0.1, 0.2, 0.3, 0.5, 1.1, 0.1});
115 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 0.7, 0.8, 1.1, 2.0});
116 m.PopulateTensor<float>(m.input2(), {0.1});
|
| topk_v2_test.cc | 38 PopulateTensor<int32_t>(top_k_, {top_k});
42 PopulateTensor<float>(input_, data);
46 PopulateTensor<uint8>(input_, data);
50 PopulateTensor<int32>(input_, data);
54 PopulateTensor<int64_t>(input_, data);
|
| bidirectional_sequence_rnn_test.cc | [all...] |
| batch_to_space_nd_test.cc | 30 PopulateTensor<float>(input_, data);
34 PopulateTensor<int>(block_shape_, data);
38 PopulateTensor<int>(crops_, data);
|
| depthwise_conv_test.cc | 82 void SetFilter(std::initializer_list<float> f) { PopulateTensor(filter_, f); }
84 void SetBias(std::initializer_list<float> f) { PopulateTensor(bias_, f); }
87 PopulateTensor(input_, data);
|
| embedding_lookup_sparse_test.cc | 56 PopulateTensor(lookup_, lookup_data);
57 PopulateTensor(indices_, indices_data);
58 PopulateTensor(dense_shape_, dense_shape_data);
59 PopulateTensor(weights_, weights_data);
|
| l2norm_test.cc | 38 PopulateTensor(input_, data);
|
| svdf_test.cc | 156 PopulateTensor(weights_feature_, f);
161 PopulateTensor(weights_time_, f);
166 PopulateTensor(input_, offset, begin, end);
174 PopulateTensor(state_, 0, zero_buffer.get(),
|
| embedding_lookup_test.cc | 44 PopulateTensor(input_, data);
73 m.PopulateTensor<int>(0, {1, 0, 2});
|
| gather_test.cc | 40 PopulateTensor<float>(input_, data);
44 PopulateTensor<uint8_t>(input_, data);
52 PopulateTensor<int>(positions_, data);
|
| space_to_batch_nd_test.cc | 30 PopulateTensor<float>(input_, data);
34 PopulateTensor<int>(block_shape_, data);
38 PopulateTensor<int>(paddings_, data);
|
| mean_test.cc | 28 void SetAxis(std::initializer_list<int> data) { PopulateTensor(axis_, data); }
32 PopulateTensor(input_, data);
|
| pad_test.cc | 29 PopulateTensor<float>(input_, data);
33 PopulateTensor<int>(paddings_, paddings);
|
| resize_bilinear_test.cc | 49 PopulateTensor(input_, data);
51 void SetSize(std::initializer_list<int> data) { PopulateTensor(size_, data); }
|
| softmax_test.cc | 44 PopulateTensor(input_, data);
48 PopulateTensor(input_, offset, begin, end);
|
| split_test.cc | 51 PopulateTensor(input_, data);
53 void SetAxis(int axis) { PopulateTensor(axis_, {axis}); }
|
| exp_test.cc | 38 PopulateTensor(input_, data);
|
| /external/tensorflow/tensorflow/contrib/lite/models/smartreply/ops/ |
| predict_test.cc | 57 PopulateTensor<int>(input_signature_, data);
61 PopulateTensor<int>(model_key_, data);
65 PopulateTensor<int>(model_label_, data);
69 PopulateTensor<float>(model_weight_, data);
|
