| /external/tensorflow/tensorflow/contrib/lite/tools/ |
| verifier_test.cc | 57 void AddTensor(const std::vector<int>& shape, tflite::TensorType type,
123 builder.AddTensor({2, 3}, TensorType_UINT8, {1, 2, 3, 4, 5, 6}, "input");
124 builder.AddTensor(
128 builder.AddTensor({2, 3}, TensorType_INT32, {}, "output");
167 builder.AddTensor({2, 3}, TensorType_UINT8, {1, 2, 3, 4}, "input");
174 builder.AddTensor({2, 1}, TensorType_UINT8, {1, 2, 3, 4}, "input");
181 builder.AddTensor({1024, 2048, 4096}, TensorType_UINT8, {1, 2, 3, 4},
213 builder.AddTensor(
223 builder.AddTensor(
232 builder.AddTensor(
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| session_state.h | 37 Status AddTensor(const string& handle, const Tensor& tensor);
71 Status AddTensor(const string& name, const TensorAndKey& tk);
|
| dataset.h | 114 Status AddTensor(const Tensor& val, Node** output) {
117 return errors::Internal("AddTensor: Failed to build Const op.");
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| session_state.cc | 34 Status SessionState::AddTensor(const string& handle, const Tensor& tensor) {
57 Status TensorStore::AddTensor(const string& name, const TensorAndKey& tk) {
78 TF_RETURN_IF_ERROR(session_state->AddTensor(key, it->second.tensor));
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| test_util.cc | 52 int SingleOpModel::AddTensor(TensorData t, std::initializer_list<int> data) {
106 int id = AddTensor(t, {});
114 int id = AddTensor(TensorData{type, shape}, data);
126 int id = AddTensor(t, {});
|
| test_util.h | 210 int AddTensor(TensorData t, std::initializer_list<int> data);
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| tensor_dataset_op.cc | 76 TF_RETURN_IF_ERROR(b->AddTensor(t, &node));
|
| tensor_slice_dataset_op.cc | 93 TF_RETURN_IF_ERROR(b->AddTensor(t, &node));
|
| scan_dataset_op.cc | 118 TF_RETURN_IF_ERROR(b->AddTensor(t, &node));
127 TF_RETURN_IF_ERROR(b->AddTensor(t, &node));
|
| sparse_tensor_slice_dataset_op.cc | 61 TF_RETURN_IF_ERROR(b->AddTensor(sparse_tensor_.indices(), &indices_node));
63 TF_RETURN_IF_ERROR(b->AddTensor(sparse_tensor_.values(), &value_node));
|
| padded_batch_dataset_op.cc | 157 TF_RETURN_IF_ERROR(b->AddTensor(t, &node));
165 TF_RETURN_IF_ERROR(b->AddTensor(t, &node));
|
| filter_dataset_op.cc | 99 TF_RETURN_IF_ERROR(b->AddTensor(t, &node));
|
| map_dataset_op.cc | 104 TF_RETURN_IF_ERROR(b->AddTensor(t, &node));
|
| flat_map_dataset_op.cc | 106 TF_RETURN_IF_ERROR(b->AddTensor(t, &node));
|
| interleave_dataset_op.cc | 130 TF_RETURN_IF_ERROR(b->AddTensor(t, &node));
|
| parallel_map_dataset_op.cc | 118 TF_RETURN_IF_ERROR(b->AddTensor(t, &node));
|
| tensor_queue_dataset_op.cc | 117 TF_RETURN_IF_ERROR(b->AddTensor(t, &node));
125 TF_RETURN_IF_ERROR(b->AddTensor(t, &node));
|
| group_by_window_dataset_op.cc | 505 TF_RETURN_IF_ERROR(b->AddTensor(t, &node));
|
| /external/tensorflow/tensorflow/core/kernels/ |
| session_ops.cc | 47 OP_REQUIRES_OK(ctx, ctx->tensor_store()->AddTensor(name(), tk));
|
