Searched refs:is_nhwc (Results 1 – 6 of 6) sorted by relevance
| /external/ComputeLibrary/src/core/CL/kernels/ |
| D | CLChannelShuffleLayerKernel.cpp | 69 const bool is_nhwc = input->data_layout() == DataLayout::NHWC; in validate_and_configure_window() local
70 if(is_nhwc) in validate_and_configure_window()
118 const bool is_nhwc = data_layout == DataLayout::NHWC; in configure() local
122 if(is_nhwc) in configure()
137 …build_opts.add_option_if(is_nhwc, "-DVEC_SIZE_LEFTOVER=" + support::cpp11::to_string(vec_size_x_le… in configure()
138 …build_opts.add_option_if(is_nhwc, "-DSRC_DIM_X=" + support::cpp11::to_string(input->info()->dimens… in configure()
|
| /external/ComputeLibrary/src/gpu/cl/kernels/ |
| D | ClScaleKernel.cpp | 100 const bool is_nhwc = _data_layout == DataLayout::NHWC; in configure() local
188 if(is_nhwc) in configure()
198 _config_id += (is_nhwc ? "nhwc" : "nchw"); in configure()
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEDepthwiseConvolutionLayer.cpp | 77 bool is_nhwc = input->info()->data_layout() == DataLayout::NCHW; in configure() local
82 _impl->permute = is_nhwc; in configure()
103 if(is_nhwc) in configure()
|
| /external/ComputeLibrary/src/graph/ |
| D | GraphBuilder.cpp | 741 const bool is_nhwc = input_tensor_desc.layout == DataLayout::NHWC; in add_yolo_node() local
747 const Coordinates box_end = is_nhwc ? Coordinates(3, -1, -1) : Coordinates(-1, -1, 3); in add_yolo_node()
756 const Coordinates imm_start = is_nhwc ? Coordinates(3, 0, 0) : Coordinates(0, 0, 3); in add_yolo_node()
757 const Coordinates imm_end = is_nhwc ? Coordinates(5, -1, -1) : Coordinates(-1, -1, 5); in add_yolo_node()
763 const Coordinates cls_start = is_nhwc ? Coordinates(5, 0, 0) : Coordinates(0, 0, 5); in add_yolo_node()
|
| /external/ComputeLibrary/tests/ |
| D | AssetsLibrary.h | 589 const bool is_nhwc = tensor.data_layout() == DataLayout::NHWC; in fill_with_generator() local
592 if(is_nhwc) in fill_with_generator()
604 if(is_nhwc) in fill_with_generator()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 4834 bool is_nhwc = RandomBool(); in TEST_F() local
4837 std::vector<int64_t> scale_dims = {x_dims[is_nhwc ? 3 : 1]}; in TEST_F()
4838 std::vector<int64_t> offset_dims = {x_dims[is_nhwc ? 3 : 1]}; in TEST_F()
4851 .Attr("data_format", is_nhwc ? "NHWC" : "NCHW") in TEST_F()
|