/aosp_15_r20/external/ComputeLibrary/examples/ |
H A D | neon_cnn.cpp | 134 constexpr auto data_layout = DataLayout::NCHW; in do_setup() local
|
/aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/ |
H A D | CLPriorBoxLayerKernel.cpp | 128 const DataLayout data_layout = input1->info()->data_layout(); in configure() local
|
/aosp_15_r20/external/ComputeLibrary/tests/validate_examples/ |
H A D | graph_convolution.cpp | 237 EnumOption<arm_compute::DataLayout> *data_layout; /**< Graph data layout */ member in __anon238537030111::ConvolutionOptions
|
H A D | graph_depthwiseconvolution.cpp | 238 …EnumOption<arm_compute::DataLayout> *data_layout; /**< Graph data la… member in __anonf2723d900111::DepthConvolutionOptions
|
/aosp_15_r20/external/ComputeLibrary/utils/ |
H A D | ImageLoader.h | 262 const DataLayout data_layout = tensor.info()->data_layout(); variable
|
H A D | TypePrinter.h | 858 inline std::string to_string(const arm_compute::DataLayout &data_layout) in to_string() 1279 const DataLayout data_layout = info->data_layout(); variable
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
H A D | dnn.proto | 101 DataLayout data_layout = 3; field
|
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/operators/ |
H A D | ClConv2d.cpp | 220 const DataLayout data_layout = std::get<4>(config); in get_convolution_method() local
|
H A D | ClWinogradConv2d.cpp | 51 …D winograd_output_tile(const Size2D &input_dims, const Size2D &kernel_dims, DataLayout data_layout) in winograd_output_tile()
|
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/ |
H A D | NEPriorBoxLayerKernel.cpp | 106 const DataLayout data_layout = _input1->info()->data_layout(); in calculate_prior_boxes() local
|
H A D | NESpaceToBatchLayerKernel.cpp | 58 const DataLayout data_layout = input->data_layout(); in validate_arguments() local
|
/aosp_15_r20/external/llvm/bindings/ocaml/target/ |
H A D | llvm_target.mli | 198 val data_layout : t -> DataLayout.t val
|
/aosp_15_r20/external/ComputeLibrary/src/cpu/operators/ |
H A D | CpuScale.cpp | 156 …const DataLayout data_layout = info.data_layout == DataLayout::UNKNOWN ? src->data_layout() : info… in validate() local
|
H A D | CpuWinogradConv2d.cpp | 58 const DataLayout data_layout = in->data_layout(); in internal_get_shape() local
|
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/ |
H A D | ClPool2dKernel.cpp | 52 …const auto data_layout = pool_info.data_layout == DataLayout::UNKNOWN ? src->data_layout()… in validate_arguments() local
|
H A D | ClDirectConv2dKernel.cpp | 59 const DataLayout data_layout = src->data_layout(); in validate_arguments() local
|
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
H A D | CLGEMMDeconvolutionLayer.cpp | 131 DataLayout data_layout = input->data_layout(); in validate() local
|
/aosp_15_r20/external/ComputeLibrary/src/core/helpers/ |
H A D | PoolingHelpers.h | 56 inline float calculate_avg_scale_pool2d(bool exclude_padding, DataLayout data_layout, const Coordin… in calculate_avg_scale_pool2d()
|
/aosp_15_r20/external/ComputeLibrary/src/core/CL/ |
H A D | CLHelpers.cpp | 290 …ution_layer_supported(const Size2D &output_tile, const Size2D &kernel_size, DataLayout data_layout) in cl_winograd_convolution_layer_supported()
|
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
H A D | FullyConnectedLayerFixture.h | 77 const DataLayout data_layout = src.info()->data_layout(); in mix_layout() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | fused_batch_norm_op.cc | 896 const se::dnn::DataLayout data_layout = in operator ()() local 1091 const se::dnn::DataLayout data_layout = in operator ()() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
H A D | vector_support_library.cc | 153 const llvm::DataLayout& data_layout = in IntegerTypeForFloatSize() local
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/ |
H A D | api.h | 176 DataLayout data_layout = DataLayout::UNKNOWN; member
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/ |
H A D | kernel_ops_pattern.cc | 223 const char* data_layout = xla::gpu::amdgpu::DataLayout(); in Emit() local
|
/aosp_15_r20/external/ComputeLibrary/src/core/ |
H A D | Utils.cpp | 367 …input_shape, TensorShape weights_shape, PadStrideInfo conv_info, DataLayout data_layout, const Siz… in calculate_same_pad()
|