| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | buffer_map_test.cc | 77 return buffer_map.GetTensor(0); in MakeTensor()
107 EXPECT_THAT(GetTensorData<float>(buffer_map.GetTensor(0)), in TEST()
111 tensorflow::Tensor out_tensor = buffer_map.GetTensor(0); in TEST()
125 EXPECT_THAT(GetTensorData<tensorflow::tstring>(buffer_map.GetTensor(0)), in TEST()
129 tensorflow::Tensor out_tensor = buffer_map.GetTensor(0); in TEST()
145 EXPECT_THAT(GetTensorData<int>(buffer_map.GetTensor(0)), in TEST()
159 EXPECT_THAT(GetTensorData<tensorflow::tstring>(buffer_map.GetTensor(0)), in TEST()
170 EXPECT_THAT(GetTensorData<float>(buffer_map.GetTensor(0)), in TEST()
174 tensorflow::Tensor out_tensor = buffer_map.GetTensor(0); in TEST()
188 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 | 108 tensorflow::Tensor t = buffer_map->GetTensor(buffer_handle); in CopyFromBufferHandle()
|
| D | buffer_map.cc | 148 tensorflow::Tensor BufferMap::GetTensor(int tensor_index) const { in GetTensor() function in tflite::flex::BufferMap
|
| /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
|
| /external/tensorflow/tensorflow/lite/experimental/resource/ |
| D | resource_variable_test.cc | 51 auto* value = var.GetTensor(); in TEST()
77 auto* value = var.GetTensor(); in TEST()
105 auto* value = var.GetTensor(); in TEST()
117 value = var.GetTensor(); in TEST()
146 auto* value = var.GetTensor(); in TEST()
158 value = var.GetTensor(); in TEST()
|
| D | resource_variable.h | 47 TfLiteTensor* GetTensor() { return is_initialized_ ? &tensor_ : nullptr; } in GetTensor() function
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | kernel_runner.cc | 45 context_.GetTensor = GetTensor; in KernelRunner()
76 TfLiteTensor* KernelRunner::GetTensor(const struct TfLiteContext* context, in GetTensor() function in tflite::micro::KernelRunner
|
| D | kernel_runner.h | 49 static TfLiteTensor* GetTensor(const struct TfLiteContext* context,
|
| /external/tensorflow/tensorflow/c/ |
| D | checkpoint_reader.cc | 88 void CheckpointReader::GetTensor( in GetTensor() function in tensorflow::checkpoint::CheckpointReader
93 status = reader_->GetTensor(name, out_tensor); in GetTensor()
|
| D | checkpoint_reader.h | 57 void GetTensor(const string& name,
|
| /external/tensorflow/tensorflow/lite/micro/ |
| D | micro_interpreter.cc | 83 TfLiteTensor* ContextHelper::GetTensor(const struct TfLiteContext* context, in GetTensor() function in tflite::internal::ContextHelper
173 context_.GetTensor = context_helper_.GetTensor; in Init()
|
| D | micro_interpreter.h | 56 static TfLiteTensor* GetTensor(const struct TfLiteContext* context,
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | session_state.h | 34 Status GetTensor(const std::string& handle, Tensor* tensor);
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | session_state.cc | 26 Status SessionState::GetTensor(const string& handle, Tensor* tensor) { in GetTensor() function in tensorflow::SessionState
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | read_variable.cc | 62 TfLiteTensor* variable_tensor = variable->GetTensor(); in Eval()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | inference_context.cc | 618 node.cl_operation.GetGpuOperation().SetSrc(GetTensor(node.inputs[i]), i); in BindMemoryToOperations()
621 node.cl_operation.GetGpuOperation().SetDst(GetTensor(node.outputs[i]), i); in BindMemoryToOperations()
700 Tensor* InferenceContext::GetTensor(ValueId id) { in GetTensor() function in tflite::gpu::cl::InferenceContext
716 return GetTensor(id)->WriteData(queue, tensor); in SetInputTensor()
722 const auto& gpu_tensor = *GetTensor(id); in GetOutputTensor()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | session_ops.cc | 104 OP_REQUIRES_OK(ctx, ctx->session_state()->GetTensor(name, &val)); in Compute()
|
| /external/tensorflow/tensorflow/python/util/ |
| D | py_checkpoint_reader_wrapper.cc | 107 reader->GetTensor(name, &tensor, status.get()); in CheckpointReader_GetTensor()
|
| /external/tensorflow/tensorflow/lite/python/interpreter_wrapper/ |
| D | interpreter_wrapper.h | 89 PyObject* GetTensor(int i) const;
|
| D | interpreter_wrapper_pybind11.cc | 142 return tensorflow::PyoOrThrow(self.GetTensor(i)); in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_reader.h | 101 Status GetTensor(const string& name,
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ethos_u/ |
| D | ethosu.cc | 56 TfLiteTensor* tensor = context->GetTensor(context, node->inputs->data[0]); in Prepare()
|