| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tflite/ |
| H A D | types.cc | 34 char* tensor_buffer; in CopyStringToBuffer() local
35 int bytes = dyn_buffer.WriteToBuffer(&tensor_buffer); in CopyStringToBuffer()
37 memcpy(dst_data.data(), tensor_buffer, bytes); in CopyStringToBuffer()
38 free(tensor_buffer); in CopyStringToBuffer()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/ |
| H A D | buffer_map_test.cc | 281 TfLiteTensorBuffer* tensor_buffer = in TEST() local
284 EXPECT_FALSE(tensor_buffer->BufferReusedFromTfLiteTensor()); in TEST()
285 EXPECT_NE(tensor_buffer->data(), tensor.data.raw); in TEST()
286 tensor_buffer->Unref(); in TEST()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/ |
| H A D | string_util.cc | 109 char* tensor_buffer; in WriteToTensor() local
110 int bytes = WriteToBuffer(&tensor_buffer); in WriteToTensor()
118 tensor_buffer, bytes, kTfLiteDynamic, tensor->allocation, in WriteToTensor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| H A D | parse_example.cc | 433 inline void CopyToBuffer(tf::gtl::ArraySlice<tstring> vec, char* tensor_buffer, in CopyToBuffer() argument
440 memcpy(tensor_buffer + start, vec[k].c_str(), vec[k].size()); in CopyToBuffer()
448 memcpy(tensor_buffer + start, vec[k].c_str(), vec[k].size()); in CopyToBuffer()
618 char* tensor_buffer = in FastParseExampleLite() local
624 tensor_buffer = reinterpret_cast<char*>(malloc(required_bytes)); in FastParseExampleLite()
625 result->dense_values[d]->data.raw = tensor_buffer; in FastParseExampleLite()
629 memcpy(tensor_buffer, &num_strings, sizeof(int32_t)); in FastParseExampleLite()
632 memcpy(tensor_buffer + sizeof(int32_t) * (i + 1), &offset_i, in FastParseExampleLite()
636 CopyToBuffer(slice, tensor_buffer + start, count, batch_size, in FastParseExampleLite()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | tensor_cord.cc | 55 void TensorCord::TensorBufReleaser(void* tensor_buffer) { in TensorBufReleaser() argument
56 static_cast<TensorBuffer*>(tensor_buffer)->Unref(); in TensorBufReleaser()
|
| H A D | tensor_cord.h | 267 static void TensorBufReleaser(void* tensor_buffer);
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| H A D | lstm_utils.cc | 73 void CopyArrayToSubArray(Buffer<ArrayDataType::kFloat>& tensor_buffer, in CopyArrayToSubArray() argument
83 CopyArrayData(sub_buffer, dim2_copy_size, 0, 0, &tensor_buffer, in CopyArrayToSubArray()
|
| H A D | lstm_utils.h | 101 void CopyArrayToSubArray(Buffer<ArrayDataType::kFloat>& tensor_buffer,
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | tensor.h | 1034 HostScalarTensorBuffer tensor_buffer; member
1053 intptr_t offset = reinterpret_cast<intptr_t>(&dummy_object->tensor_buffer) - in delete()
1065 new (&value_and_buf->tensor_buffer) in Tensor()
1068 buf_ = &value_and_buf->tensor_buffer; in Tensor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | xla_launch_util.cc | 310 auto* tensor_buffer = new XlaTensorBuffer(buffer.opaque(), expected_size, in MakeTensor() local
312 Tensor t(dtype, shape, tensor_buffer); in MakeTensor()
313 tensor_buffer->Unref(); in MakeTensor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | snapshot_utils.cc | 249 auto tensor_buffer = DMAHelper::buffer(&tensor); in WriteTensors() local
250 tensor_buffers.push_back(tensor_buffer); in WriteTensors()
251 size = tensor_buffer->size(); in WriteTensors()
|
| /aosp_15_r20/external/pytorch/torch/distributed/fsdp/ |
| H A D | _optim_utils.py | 243 tensor_buffer = value.new_zeros(*buffer_size)
245 tensor_buffer, value, group=fsdp_state.process_group
251 tensor_state[state_name] = tensor_buffer[:unpadded_numel]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ |
| H A D | flatbuffer_export.cc | 779 char* tensor_buffer; in BuildBuffer() local
780 int bytes = dynamic_buffer.WriteToBuffer(&tensor_buffer); in BuildBuffer()
782 builder_.CreateVector(reinterpret_cast<uint8_t*>(tensor_buffer), bytes); in BuildBuffer()
783 free(tensor_buffer); in BuildBuffer()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/ |
| H A D | quantize_model_test.cc | 2218 std::vector<float> tensor_buffer = {0.0, 5.0}; in MakeInputConstant() local
2219 uint8_t* uint8_data = reinterpret_cast<uint8_t*>(tensor_buffer.data()); in MakeInputConstant()
2221 uint8_data, uint8_data + (sizeof(float) * tensor_buffer.size())); in MakeInputConstant()
|