| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | buffer_map_test.cc | 83 return buffer_map.GetTensor(0); in MakeTensor()
113 EXPECT_THAT(GetTensorData<float>(buffer_map.GetTensor(0)), in TEST()
117 tensorflow::Tensor out_tensor = buffer_map.GetTensor(0); in TEST()
131 EXPECT_THAT(GetTensorData<string>(buffer_map.GetTensor(0)), in TEST()
135 tensorflow::Tensor out_tensor = buffer_map.GetTensor(0); in TEST()
151 EXPECT_THAT(GetTensorData<int>(buffer_map.GetTensor(0)), in TEST()
165 EXPECT_THAT(GetTensorData<string>(buffer_map.GetTensor(0)), in TEST()
176 EXPECT_THAT(GetTensorData<float>(buffer_map.GetTensor(0)), in TEST()
180 tensorflow::Tensor out_tensor = buffer_map.GetTensor(0); in TEST()
194 EXPECT_THAT(GetTensorData<int>(buffer_map.GetTensor(0)), in TEST()
[all …]
|
| D | buffer_map.h | 48 tensorflow::Tensor GetTensor(int tensor_index) const;
|
| D | delegate.cc | 87 tensorflow::Tensor t = buffer_map->GetTensor(buffer_handle); in CopyFromBufferHandle()
|
| D | buffer_map.cc | 143 tensorflow::Tensor BufferMap::GetTensor(int tensor_index) const { in GetTensor() function in tflite::flex::BufferMap
|
| D | kernel.cc | 288 buffer_map->GetTensor(input_index), nullptr, nullptr, nullptr); in BuildEagerInputs()
550 CopyShapeAndType(context, buffer_map->GetTensor(tensor_index), tensor)); in Eval()
|
| D | kernel_test.cc | 89 buffer_map->GetTensor(buffer_handle).tensor_data(); in ConfigureDelegate()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | cc_ops_test.cc | 53 test::GetTensor(root, m, &out); in TEST()
62 test::GetTensor(root, m, &out); in TEST()
72 test::GetTensor(root, c, &out); in TEST()
83 test::GetTensor(root, l, &out); in TEST()
114 test::GetTensor(root, y_normalized, &out); in TEST()
199 test::GetTensor(root, c1, &out); in TEST()
204 test::GetTensor(root, c2, &out); in TEST()
216 test::GetTensor(root, c1, &out); in TEST()
221 test::GetTensor(root, c2, &out); in TEST()
226 test::GetTensor(root, c3, &out); in TEST()
[all …]
|
| D | testutil.h | 36 void GetTensor(const Scope& scope, Output tensor, Tensor* out);
41 void GetTensor(const Scope& scope, const std::vector<Output>& assign_vars,
|
| D | testutil.cc | 33 void GetTensor(const Scope& scope, Output tensor, Tensor* out) { in GetTensor() function
46 void GetTensor(const Scope& scope, const std::vector<Output>& assign_vars, in GetTensor() function
|
| D | gradients_test.cc | 544 test::GetTensor(scope_, grad_outputs[0], &output); in CheckGrad()
621 test::GetTensor(scope_, grad_outputs[0], &output); in CheckGrad()
|
| /external/tensorflow/tensorflow/c/ |
| D | checkpoint_reader.cc | 89 void CheckpointReader::GetTensor( in GetTensor() function in tensorflow::checkpoint::CheckpointReader
94 status = reader_->GetTensor(name, out_tensor); in GetTensor()
|
| D | checkpoint_reader.h | 57 void GetTensor(const string& name,
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | session_state.h | 34 Status GetTensor(const string& handle, Tensor* tensor);
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | session_state.cc | 23 Status SessionState::GetTensor(const string& handle, Tensor* tensor) { in GetTensor() function in tensorflow::SessionState
|
| D | direct_session.cc | 1026 return session_state_.GetTensor(resource_handle.name(), retrieved_tensor); in ResourceHandleToInputTensor()
|
| /external/tensorflow/tensorflow/lite/python/interpreter_wrapper/ |
| D | interpreter_wrapper.h | 68 PyObject* GetTensor(int i) const;
|
| D | interpreter_wrapper.cc | 329 PyObject* InterpreterWrapper::GetTensor(int i) const { in GetTensor() function in tflite::interpreter_wrapper::InterpreterWrapper
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | interpreter.py | 229 return self._interpreter.GetTensor(tensor_index)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | session_ops.cc | 115 OP_REQUIRES_OK(ctx, ctx->session_state()->GetTensor(name, &val)); in Compute()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_reader.h | 101 Status GetTensor(const string& name,
|
| D | tensor_slice_reader.cc | 224 Status TensorSliceReader::GetTensor( in GetTensor() function in tensorflow::checkpoint::TensorSliceReader
|
| /external/tensorflow/tensorflow/python/util/ |
| D | py_checkpoint_reader.i | 110 reader->GetTensor(name, &tensor, out_status); in CheckpointReader_GetTensor()
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_show.cc | 60 ckpt_reader_->GetTensor(name, &out_tensor, status); in LookUpCheckPoint()
|