Home
last modified time | relevance | path

Searched defs:GetTensor (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/cc/framework/
Dtestutil.cc33 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/
Dresource_variable.h47 TfLiteTensor* GetTensor() { return is_initialized_ ? &tensor_ : nullptr; } in GetTensor() function
/external/tensorflow/tensorflow/c/
Dcheckpoint_reader.cc88 void CheckpointReader::GetTensor( in GetTensor() function in tensorflow::checkpoint::CheckpointReader
/external/tensorflow/tensorflow/core/common_runtime/
Dsession_state.cc26 Status SessionState::GetTensor(const string& handle, Tensor* tensor) { in GetTensor() function in tensorflow::SessionState
/external/tensorflow/tensorflow/lite/micro/kernels/
Dkernel_runner.cc76 TfLiteTensor* KernelRunner::GetTensor(const struct TfLiteContext* context, in GetTensor() function in tflite::micro::KernelRunner
/external/tensorflow/tensorflow/lite/delegates/flex/
Dbuffer_map.cc148 tensorflow::Tensor BufferMap::GetTensor(int tensor_index) const { in GetTensor() function in tflite::flex::BufferMap
/external/tensorflow/tensorflow/core/util/
Dtensor_slice_reader.cc229 Status TensorSliceReader::GetTensor( in GetTensor() function in tensorflow::checkpoint::TensorSliceReader
/external/tensorflow/tensorflow/lite/micro/
Dmicro_interpreter.cc83 TfLiteTensor* ContextHelper::GetTensor(const struct TfLiteContext* context, in GetTensor() function in tflite::internal::ContextHelper
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h183 const Tensor& GetTensor() const { return tensor_; } in GetTensor() function
/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/
Dinterpreter_wrapper.cc610 PyObject* InterpreterWrapper::GetTensor(int i) const { in GetTensor() function in tflite::interpreter_wrapper::InterpreterWrapper
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dinference_context.cc700 Tensor* InferenceContext::GetTensor(ValueId id) { in GetTensor() function in tflite::gpu::cl::InferenceContext
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dinference_context.cc384 MetalSpatialTensor* InferenceContext::GetTensor(ValueId tensor_id) { in GetTensor() function in tflite::gpu::metal::InferenceContext
/external/tensorflow/tensorflow/lite/c/
Dcommon.h760 TfLiteTensor* (*GetTensor)(const struct TfLiteContext* context, member