Home
last modified time | relevance | path

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

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