Home
last modified time | relevance | path

Searched refs:data_f (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dmetal_spatial_tensor.cc390 std::unique_ptr<float[]> data_f; in WriteDataBHWDC() local
393 data_f.reset(new float[elements_count]); in WriteDataBHWDC()
394 data_ptr = data_f.get(); in WriteDataBHWDC()
395 DataFromBHWDC(in, shape_, descriptor_, data_f.get()); in WriteDataBHWDC()
455 std::unique_ptr<float[]> data_f; in ReadDataBHWDC() local
458 data_f.reset(new float[elements_count]); in ReadDataBHWDC()
459 data_ptr = data_f.get(); in ReadDataBHWDC()
485 DataToBHWDC(data_f.get(), shape_, descriptor_, out); in ReadDataBHWDC()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dtensor.cc494 std::unique_ptr<float[]> data_f; in WriteDataBHWDC() local
497 data_f.reset(new float[elements_count]); in WriteDataBHWDC()
498 data_ptr = data_f.get(); in WriteDataBHWDC()
499 DataFromBHWDC(in, shape_, descriptor_, data_f.get()); in WriteDataBHWDC()
555 std::unique_ptr<float[]> data_f; in ReadDataBHWDC() local
558 data_f.reset(new float[elements_count]); in ReadDataBHWDC()
559 data_ptr = data_f.get(); in ReadDataBHWDC()
582 DataToBHWDC(data_f.get(), shape_, descriptor_, out); in ReadDataBHWDC()