Searched refs:data_h (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | metal_spatial_tensor.cc | 391 std::unique_ptr<half[]> data_h; in WriteDataBHWDC() local
397 data_h.reset(new half[elements_count]); in WriteDataBHWDC()
398 data_ptr = data_h.get(); in WriteDataBHWDC()
399 DataFromBHWDC(in, shape_, descriptor_, data_h.get()); in WriteDataBHWDC()
456 std::unique_ptr<half[]> data_h; in ReadDataBHWDC() local
461 data_h.reset(new half[elements_count]); in ReadDataBHWDC()
462 data_ptr = data_h.get(); in ReadDataBHWDC()
487 DataToBHWDC(data_h.get(), shape_, descriptor_, out); in ReadDataBHWDC()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | tensor.cc | 495 std::unique_ptr<half[]> data_h; in WriteDataBHWDC() local
501 data_h.reset(new half[elements_count]); in WriteDataBHWDC()
502 data_ptr = data_h.get(); in WriteDataBHWDC()
503 DataFromBHWDC(in, shape_, descriptor_, data_h.get()); in WriteDataBHWDC()
556 std::unique_ptr<half[]> data_h; in ReadDataBHWDC() local
561 data_h.reset(new half[elements_count]); in ReadDataBHWDC()
562 data_ptr = data_h.get(); in ReadDataBHWDC()
584 DataToBHWDC(data_h.get(), shape_, descriptor_, out); in ReadDataBHWDC()
|