Searched refs:tensor_buffer (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/ |
| D | string_util.cc | 107 char* tensor_buffer; in WriteToTensor() local
108 int bytes = WriteToBuffer(&tensor_buffer); in WriteToTensor()
116 tensor_buffer, bytes, kTfLiteDynamic, tensor->allocation, in WriteToTensor()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.h | 224 auto* tensor_buffer = new XlaTensorBuffer(buffer.opaque(), expected_size, in MakeTensor() local
226 Tensor t(dtype, shape, tensor_buffer); in MakeTensor()
227 tensor_buffer->Unref(); in MakeTensor()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | types.cc | 31 char* tensor_buffer; in CopyStringToBuffer() local
32 int bytes = dyn_buffer.WriteToBuffer(&tensor_buffer); in CopyStringToBuffer()
34 memcpy(dst_data.data(), tensor_buffer, bytes); in CopyStringToBuffer()
35 free(tensor_buffer); in CopyStringToBuffer()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | lstm_utils.cc | 71 void CopyArrayToSubArray(Buffer<ArrayDataType::kFloat>& tensor_buffer, in CopyArrayToSubArray() argument
81 CopyArrayData(sub_buffer, dim2_copy_size, 0, 0, &tensor_buffer, in CopyArrayToSubArray()
|
| D | lstm_utils.h | 101 void CopyArrayToSubArray(Buffer<ArrayDataType::kFloat>& tensor_buffer,
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.h | 959 HostScalarTensorBuffer tensor_buffer; member
978 intptr_t offset = reinterpret_cast<intptr_t>(&dummy_object->tensor_buffer) - in delete()
990 new (&value_and_buf->tensor_buffer) in Tensor()
993 buf_ = &value_and_buf->tensor_buffer; in Tensor()
|