Home
last modified time | relevance | path

Searched defs:tensor_buffer (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtensor_cord.cc55 void TensorCord::TensorBufReleaser(void* tensor_buffer) { in TensorBufReleaser()
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dstring_util.cc109 char* tensor_buffer; in WriteToTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tflite/
H A Dtypes.cc34 char* tensor_buffer; in CopyStringToBuffer() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dlstm_utils.cc73 void CopyArrayToSubArray(Buffer<ArrayDataType::kFloat>& tensor_buffer, in CopyArrayToSubArray()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/parse_example/
H A Dparse_example.cc433 inline void CopyToBuffer(tf::gtl::ArraySlice<tstring> vec, char* tensor_buffer, in CopyToBuffer()
618 char* tensor_buffer = in FastParseExampleLite() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Dbuffer_map_test.cc281 TfLiteTensorBuffer* tensor_buffer = in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dtensor.h1034 HostScalarTensorBuffer tensor_buffer; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_launch_util.cc310 auto* tensor_buffer = new XlaTensorBuffer(buffer.opaque(), expected_size, in MakeTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dsnapshot_utils.cc249 auto tensor_buffer = DMAHelper::buffer(&tensor); in WriteTensors() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/
H A Dflatbuffer_export.cc779 char* tensor_buffer; in BuildBuffer() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/
H A Dquantize_model_test.cc2218 std::vector<float> tensor_buffer = {0.0, 5.0}; in MakeInputConstant() local