Home
last modified time | relevance | path

Searched refs:DHWC4 (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dtensor_type_util.cc40 return DataLayout::DHWC4; in ToDataLayout()
42 return DataLayout::DHWC4; in ToDataLayout()
48 return DataLayout::DHWC4; in ToDataLayout()
50 return DataLayout::DHWC4; in ToDataLayout()
65 case DataLayout::DHWC4: in ToTensorStorageType()
Dgpu_api_delegate.cc67 return DataLayout::DHWC4; in ToDataLayoutFromTFL()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dconverter_test.cc73 input_def.object_def.data_layout = DataLayout::DHWC4; in RunFromTensorTest()
134 output_def.object_def.data_layout = DataLayout::DHWC4; in RunToTensorTest()
Dconverter.cc103 input.data_layout == DataLayout::DHWC4; in IsSupported()
191 output.data_layout == DataLayout::DHWC4; in IsSupported()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
Dapi2.cc173 internal_def.data_layout == DataLayout::DHWC4 && in IsSameDef()
341 inner_def.internal_def.object_def.data_layout = DataLayout::DHWC4; in MakeOuterInnerDefs()
622 external_def.object_def.data_layout = DataLayout::DHWC4; in LinkTensors()
/external/tensorflow/tensorflow/lite/delegates/gpu/
Dapi.cc112 case DataLayout::DHWC4: in NumElements()
Dapi.h67 DHWC4, enumerator
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dconverter.cc85 def.data_layout == DataLayout::DHWC4; in IsOpenCLTensor()
91 def.data_layout == DataLayout::DHWC4; in IsOpenCLTensor()