| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| optional_tensor_test.cc | 41 input_ = AddInput(TensorType_FLOAT32);
46 input_to_input_weights_ = AddInput(TensorType_FLOAT32);
49 input_to_forget_weights_ = AddInput(TensorType_FLOAT32);
50 input_to_cell_weights_ = AddInput(TensorType_FLOAT32);
51 input_to_output_weights_ = AddInput(TensorType_FLOAT32);
56 recurrent_to_input_weights_ = AddInput(TensorType_FLOAT32);
59 recurrent_to_forget_weights_ = AddInput(TensorType_FLOAT32);
60 recurrent_to_cell_weights_ = AddInput(TensorType_FLOAT32);
61 recurrent_to_output_weights_ = AddInput(TensorType_FLOAT32);
67 cell_to_input_weights_ = AddInput(TensorType_FLOAT32)
[all...] |
| lstm_test.cc | 43 input_ = AddInput(TensorType_FLOAT32);
48 input_to_input_weights_ = AddInput(TensorType_FLOAT32);
51 input_to_forget_weights_ = AddInput(TensorType_FLOAT32);
52 input_to_cell_weights_ = AddInput(TensorType_FLOAT32);
53 input_to_output_weights_ = AddInput(TensorType_FLOAT32);
58 recurrent_to_input_weights_ = AddInput(TensorType_FLOAT32);
61 recurrent_to_forget_weights_ = AddInput(TensorType_FLOAT32);
62 recurrent_to_cell_weights_ = AddInput(TensorType_FLOAT32);
63 recurrent_to_output_weights_ = AddInput(TensorType_FLOAT32);
69 cell_to_input_weights_ = AddInput(TensorType_FLOAT32)
[all...] |
| unidirectional_sequence_lstm_test.cc | 46 input_ = AddInput(TensorType_FLOAT32);
51 input_to_input_weights_ = AddInput(TensorType_FLOAT32);
54 input_to_forget_weights_ = AddInput(TensorType_FLOAT32);
55 input_to_cell_weights_ = AddInput(TensorType_FLOAT32);
56 input_to_output_weights_ = AddInput(TensorType_FLOAT32);
61 recurrent_to_input_weights_ = AddInput(TensorType_FLOAT32);
64 recurrent_to_forget_weights_ = AddInput(TensorType_FLOAT32);
65 recurrent_to_cell_weights_ = AddInput(TensorType_FLOAT32);
66 recurrent_to_output_weights_ = AddInput(TensorType_FLOAT32);
72 cell_to_input_weights_ = AddInput(TensorType_FLOAT32)
[all...] |
| lsh_projection_test.cc | 35 hash_ = AddInput(TensorType_FLOAT32);
36 input_ = AddInput(TensorType_INT32);
38 weight_ = AddInput(TensorType_FLOAT32);
|
| basic_rnn_test.cc | 129 input_ = AddInput(TensorType_FLOAT32);
130 weights_ = AddInput(TensorType_FLOAT32);
131 recurrent_weights_ = AddInput(TensorType_FLOAT32);
132 bias_ = AddInput(TensorType_FLOAT32);
|
| batch_to_space_nd_test.cc | 62 input_ = AddInput(TensorType_FLOAT32);
85 input_ = AddInput(TensorType_FLOAT32);
86 block_shape_ = AddInput(TensorType_INT32);
87 crops_ = AddInput(TensorType_INT32);
|
| depthwise_conv_test.cc | 34 input_ = AddInput(input);
35 filter_ = AddInput(filter);
39 bias_ = AddInput({TensorType_FLOAT32, {bias_size}});
46 bias_ = AddInput(bias);
|
| embedding_lookup_sparse_test.cc | 39 lookup_ = AddInput(TensorType_INT32);
40 indices_ = AddInput(TensorType_INT32);
41 dense_shape_ = AddInput(TensorType_INT32);
42 weights_ = AddInput(TensorType_FLOAT32);
43 value_ = AddInput(TensorType_FLOAT32);
|
| mean_test.cc | 54 input_ = AddInput(input);
68 input_ = AddInput(input);
69 axis_ = AddInput(axis);
|
| pad_test.cc | 56 input_ = AddInput(TensorType_FLOAT32);
77 input_ = AddInput(TensorType_FLOAT32);
78 paddings_ = AddInput(TensorType_INT32);
|
| l2norm_test.cc | 30 input_ = AddInput(TensorType_FLOAT32);
|
| space_to_batch_nd_test.cc | 62 input_ = AddInput(TensorType_FLOAT32);
85 input_ = AddInput(TensorType_FLOAT32);
86 block_shape_ = AddInput(TensorType_INT32);
87 paddings_ = AddInput(TensorType_INT32);
|
| unidirectional_sequence_rnn_test.cc | 133 input_ = AddInput(TensorType_FLOAT32);
134 weights_ = AddInput(TensorType_FLOAT32);
135 recurrent_weights_ = AddInput(TensorType_FLOAT32);
136 bias_ = AddInput(TensorType_FLOAT32);
|
| embedding_lookup_test.cc | 36 input_ = AddInput(TensorType_INT32);
37 weight_ = AddInput(TensorType_FLOAT32);
|
| bidirectional_sequence_rnn_test.cc | [all...] |
| hashtable_lookup_test.cc | 39 lookup_ = AddInput(TensorType_INT32);
40 key_ = AddInput(TensorType_INT32);
41 value_ = AddInput(type);
|
| svdf_test.cc | 137 input_ = AddInput(TensorType_FLOAT32);
138 weights_feature_ = AddInput(TensorType_FLOAT32);
139 weights_time_ = AddInput(TensorType_FLOAT32);
|
| resize_bilinear_test.cc | 31 input_ = AddInput(input);
35 size_ = AddInput({TensorType_INT32, {2}});
|
| split_test.cc | 33 axis_ = AddInput({TensorType_INT32, {1}});
37 input_ = AddInput(input);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| save_v2_op_test.cc | 62 AddInput<string>(TensorShape({}),
66 AddInput<string>(TensorShape({13}),
70 AddInput<string>(TensorShape({13}), [&tensornames](int x) -> string {
75 AddInput<bool>(TensorShape({2}), [](int x) -> bool { return x != 0; });
78 AddInput<int32>(TensorShape({10}), [](int x) -> int32 { return x + 1; });
81 AddInput<float>(TensorShape({2, 4}),
85 AddInput<double>(TensorShape({2, 4}),
89 AddInput<qint8>(TensorShape({3, 2}),
93 AddInput<qint32>(TensorShape({2, 3}), [](int x) -> qint32 {
98 AddInput<uint8>(TensorShape({11}), [](int x) -> uint8 { return x + 1; })
[all...] |
| save_op_test.cc | 68 AddInput<string>(TensorShape({}),
72 AddInput<string>(TensorShape({14}),
76 AddInput<bool>(TensorShape({2}), [](int x) -> bool { return x != 0; });
79 AddInput<int32>(TensorShape({10}), [](int x) -> int32 { return x + 1; });
82 AddInput<float>(TensorShape({2, 4}),
86 AddInput<double>(TensorShape({2, 4}),
90 AddInput<qint8>(TensorShape({3, 2}),
94 AddInput<qint32>(TensorShape({2, 3}), [](int x) -> qint32 {
99 AddInput<uint8>(TensorShape({11}), [](int x) -> uint8 { return x + 1; });
102 AddInput<int8>(TensorShape({7}), [](int x) -> int8 { return x - 7; })
[all...] |
| ops_testutil.h | 116 void AddInput(const TensorShape& shape, std::function<T(int)> input_mapping) {
117 test::FillFn(AddInput(DataTypeToEnum<T>::v(), shape), input_mapping);
120 // Like AddInput but takes in an explicit arrayslice of data.
124 test::FillValues<T>(AddInput(DataTypeToEnum<T>::v(), shape), data);
132 test::FillValues<T>(AddInput(DataTypeToEnum<T>::v(), shape), data);
209 Tensor* AddInput(DataType dtype, const TensorShape& shape) {
|
| merge_v2_checkpoints_op_test.cc | 82 AddInput<string>(TensorShape({2}),
85 AddInput<string>(TensorShape({}), [kMergedPrefix](int unused) -> string {
|
| /art/compiler/optimizing/ |
| induction_var_analysis_test.cc | 119 basic_[d]->AddInput(constant0_);
120 basic_[d]->AddInput(increment_[d]);
314 k_header->AddInput(constant0_);
322 k_header->AddInput(sub);
343 k_header->AddInput(constant0_);
352 k_body->AddInput(inc1);
356 k_body->AddInput(inc2);
359 k_header->AddInput(k_body);
385 k->AddInput(inc1);
389 k->AddInput(inc2)
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/models/smartreply/ops/ |
| predict_test.cc | 41 input_signature_ = AddInput(TensorType_INT32);
42 model_key_ = AddInput(TensorType_INT32);
43 model_label_ = AddInput(TensorType_INT32);
44 model_weight_ = AddInput(TensorType_FLOAT32);
|
