Home
last modified time | relevance | path

Searched defs:data_layout (Results 126 – 150 of 159) sorted by relevance

1234567

/aosp_15_r20/external/ComputeLibrary/examples/
H A Dneon_cnn.cpp134 constexpr auto data_layout = DataLayout::NCHW; in do_setup() local
/aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/
H A DCLPriorBoxLayerKernel.cpp128 const DataLayout data_layout = input1->info()->data_layout(); in configure() local
/aosp_15_r20/external/ComputeLibrary/tests/validate_examples/
H A Dgraph_convolution.cpp237 EnumOption<arm_compute::DataLayout> *data_layout; /**< Graph data layout */ member in __anon238537030111::ConvolutionOptions
H A Dgraph_depthwiseconvolution.cpp238 …EnumOption<arm_compute::DataLayout> *data_layout; /**< Graph data la… member in __anonf2723d900111::DepthConvolutionOptions
/aosp_15_r20/external/ComputeLibrary/utils/
H A DImageLoader.h262 const DataLayout data_layout = tensor.info()->data_layout(); variable
H A DTypePrinter.h858 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 Ddnn.proto101 DataLayout data_layout = 3; field
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/operators/
H A DClConv2d.cpp220 const DataLayout data_layout = std::get<4>(config); in get_convolution_method() local
H A DClWinogradConv2d.cpp51 …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 DNEPriorBoxLayerKernel.cpp106 const DataLayout data_layout = _input1->info()->data_layout(); in calculate_prior_boxes() local
H A DNESpaceToBatchLayerKernel.cpp58 const DataLayout data_layout = input->data_layout(); in validate_arguments() local
/aosp_15_r20/external/llvm/bindings/ocaml/target/
H A Dllvm_target.mli198 val data_layout : t -> DataLayout.t val
/aosp_15_r20/external/ComputeLibrary/src/cpu/operators/
H A DCpuScale.cpp156 …const DataLayout data_layout = info.data_layout == DataLayout::UNKNOWN ? src->data_layout() : info… in validate() local
H A DCpuWinogradConv2d.cpp58 const DataLayout data_layout = in->data_layout(); in internal_get_shape() local
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/
H A DClPool2dKernel.cpp52 …const auto data_layout = pool_info.data_layout == DataLayout::UNKNOWN ? src->data_layout()… in validate_arguments() local
H A DClDirectConv2dKernel.cpp59 const DataLayout data_layout = src->data_layout(); in validate_arguments() local
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/
H A DCLGEMMDeconvolutionLayer.cpp131 DataLayout data_layout = input->data_layout(); in validate() local
/aosp_15_r20/external/ComputeLibrary/src/core/helpers/
H A DPoolingHelpers.h56 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 DCLHelpers.cpp290 …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 DFullyConnectedLayerFixture.h77 const DataLayout data_layout = src.info()->data_layout(); in mix_layout() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfused_batch_norm_op.cc896 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 Dvector_support_library.cc153 const llvm::DataLayout& data_layout = in IntegerTypeForFloatSize() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/
H A Dapi.h176 DataLayout data_layout = DataLayout::UNKNOWN; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/
H A Dkernel_ops_pattern.cc223 const char* data_layout = xla::gpu::amdgpu::DataLayout(); in Emit() local
/aosp_15_r20/external/ComputeLibrary/src/core/
H A DUtils.cpp367 …input_shape, TensorShape weights_shape, PadStrideInfo conv_info, DataLayout data_layout, const Siz… in calculate_same_pad()

1234567