Home
last modified time | relevance | path

Searched refs:wei_shape (Results 1 – 4 of 4) sorted by relevance

/external/ComputeLibrary/src/runtime/heuristics/direct_conv/
DClDirectConvDefaultConfigValhall.cpp80 const TensorShape wei_shape = wei->tensor_shape(); in configure_G78_f32() local
86 const bool is_pointwise = (wei_shape[1] == wei_shape[2]) && wei_shape[1] == 1; in configure_G78_f32()
141 const TensorShape wei_shape = wei->tensor_shape(); in configure_G78_f16() local
147 const int32_t k = wei_shape[0]; in configure_G78_f16()
148 const bool is_pointwise = (wei_shape[1] == wei_shape[2]) && wei_shape[1] == 1; in configure_G78_f16()
277 const TensorShape wei_shape = wei->tensor_shape(); in configure_G57_f32() local
282 const bool is_pointwise = (wei_shape[1] == wei_shape[2]) && wei_shape[1] == 1; in configure_G57_f32()
337 const TensorShape wei_shape = wei->tensor_shape(); in configure_G57_f16() local
343 const bool is_pointwise = (wei_shape[1] == wei_shape[2]) && wei_shape[1] == 1; in configure_G57_f16()
/external/ComputeLibrary/src/runtime/heuristics/dwc_native/
DClDWCNativeDefaultConfigValhall.cpp80 const TensorShape wei_shape = wei->tensor_shape(); in configure_G78_f32() local
81 const size_t kernel_c = wei_shape[idx_c]; in configure_G78_f32()
82 const size_t kernel_w = wei_shape[idx_w]; in configure_G78_f32()
144 const TensorShape wei_shape = wei->tensor_shape(); in configure_G78_f16() local
146 const size_t kernel_c = wei_shape[idx_c]; in configure_G78_f16()
147 const size_t kernel_w = wei_shape[idx_w]; in configure_G78_f16()
246 const TensorShape wei_shape = wei->tensor_shape(); in configure_G77_f16() local
247 const size_t kernel_c = wei_shape[idx_c]; in configure_G77_f16()
248 const size_t kernel_w = wei_shape[idx_w]; in configure_G77_f16()
DClDWCNativeDefaultConfigBifrost.cpp47 const TensorShape wei_shape = wei->tensor_shape(); in configure_f32() local
48 const size_t kernel_c = wei_shape[idx_c]; in configure_f32()
49 const size_t kernel_w = wei_shape[idx_w]; in configure_f32()
119 const TensorShape wei_shape = wei->tensor_shape(); in configure_f16() local
121 const size_t kernel_c = wei_shape[idx_c]; in configure_f16()
122 const size_t kernel_w = wei_shape[idx_w]; in configure_f16()
/external/ComputeLibrary/src/runtime/heuristics/indirect_conv/
DClIndirectConvDefaultConfigValhall.cpp109 const TensorShape wei_shape = wei->tensor_shape(); in configure_G77_f16() local
115 const int32_t k = wei_shape[0]; in configure_G77_f16()