| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| conv_test.cc | 50 input_ = AddInput(input); 51 filter_ = AddInput(filter); 55 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); 62 bias_ = AddInput(bias);
|
| exp_test.cc | 29 input_ = AddInput(input);
|
| fully_connected_test.cc | 131 input_ = AddInput(input); 133 AddInput({input.type, {units_, input_size_}, input.min, input.max}); 136 bias_ = AddInput({TensorType_FLOAT32, {units_}}); 143 bias_ = AddInput(bias);
|
| local_response_norm_test.cc | 30 input_ = AddInput(TensorType_FLOAT32);
|
| reshape_test.cc | 30 input_ = AddInput(TensorType_FLOAT32);
|
| gather_test.cc | 31 input_ = AddInput(input_type); 32 positions_ = AddInput(TensorType_INT32);
|
| mul_test.cc | 31 input1_ = AddInput(input1); 32 input2_ = AddInput(input2);
|
| topk_v2_test.cc | 32 input_ = AddInput(input_type); 33 top_k_ = AddInput(TensorType_INT32);
|
| space_to_depth_test.cc | 30 input_ = AddInput(tensor_data);
|
| squeeze_test.cc | 31 input_ = AddInput(input);
|
| test_util.h | 64 // int a = m.AddInput({TensorType_FLOAT32, a_shape}); 65 // int b = m.AddInput({TensorType_FLOAT32, b_shape}); 115 int AddInput(TensorType type) { return AddInput(TensorData{type}); } 116 int AddInput(const TensorData& t);
|
| transpose_test.cc | 158 input_ = AddInput(TensorType_FLOAT32); 178 input_ = AddInput(TensorType_FLOAT32); 179 perm_ = AddInput(TensorType_INT32);
|
| pooling_test.cc | 34 input_ = AddInput(input);
|
| softmax_test.cc | 36 input_ = AddInput(TensorType_FLOAT32);
|
| activations_test.cc | 32 input_ = AddInput(input); 44 input_ = AddInput(input);
|
| add_test.cc | 31 input1_ = AddInput(input1); 32 input2_ = AddInput(input2);
|
| /external/tensorflow/tensorflow/contrib/lite/models/smartreply/ops/ |
| normalize_test.cc | 38 input_ = AddInput(TensorType_STRING);
|
| extract_feature_test.cc | 38 input_ = AddInput(TensorType_STRING);
|
| /art/compiler/optimizing/ |
| loop_optimization_test.cc | 223 phi->AddInput(add); 224 phi->AddInput(parameter_); 290 header_phi->AddInput(body_add); 291 header_phi->AddInput(preheader0_add); 292 header_phi->AddInput(preheader1_add);
|
| bounds_check_elimination_test.cc | 407 phi->AddInput(constant_initial); 422 phi->AddInput(add); 523 phi->AddInput(array_length); 539 phi->AddInput(add); 631 phi->AddInput(constant_initial); 645 phi->AddInput(add); 733 phi->AddInput(constant_initial); 752 phi->AddInput(add); [all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| lrn_op_test.cc | 104 AddInput<float>(TensorShape({1, 1, 1, 96}), 140 AddInput<float>(TensorShape({1, 1, 1, 16}), 183 AddInput<float>(TensorShape({BATCH, 1, 1, DEPTH}), \
|
| restore_op_test.cc | 191 AddInput<string>(TensorShape({}), 193 AddInput<string>(TensorShape({}), 435 AddInput<string>(TensorShape({}), 438 AddInput<string>(TensorShape({}), 441 AddInput<string>(TensorShape({}), [&shape_and_slice](int x) -> string {
|
| /external/tensorflow/tensorflow/core/framework/ |
| node_def_builder.cc | 109 AddInput(src_node, src_index); 130 AddInput(node_out.node, node_out.index); 164 void NodeDefBuilder::AddInput(StringPiece src_node, int src_index) {
|
| /external/tensorflow/tensorflow/core/graph/ |
| graph_partition.cc | 157 void AddInput(NodeDef* dst, StringPiece src_name, int src_slot) { [all...] |
| /frameworks/ml/nn/common/operations/ |
| SVDFTest.cpp | 204 #define AddInput(X) \ 208 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(AddInput); 210 #undef AddInput
|