Searched refs:element_size_in_bytes (Results 1 – 2 of 2) sorted by relevance
46 int element_size_in_bytes = input_nhwc->info()->element_size(); in run_op() local49 const size_t input_row_stride = src_strides[height_idx] / element_size_in_bytes; in run_op()50 const size_t input_col_stride = src_strides[width_idx] / element_size_in_bytes; in run_op()51 const size_t input_batch_stride = src_strides[batch_idx] / element_size_in_bytes; in run_op()85 const int element_size_in_bytes = dst_nhwc->info()->element_size(); in run_op() local88 const size_t out_row_stride = dst_strides[height_idx] / element_size_in_bytes; in run_op()89 const size_t out_col_stride = dst_strides[width_idx] / element_size_in_bytes; in run_op()90 const size_t out_batch_stride = dst_strides[batch_idx] / element_size_in_bytes; in run_op()
384 const int element_size_in_bytes = permuted_weights.get()->info()->element_size(); in prepare() local390 …tride = permuted_weights.get()->info()->strides_in_bytes()[height_idx] / element_size_in_bytes; in prepare()391 …stride = permuted_weights.get()->info()->strides_in_bytes()[width_idx] / element_size_in_bytes; in prepare()392 …l_stride = permuted_weights.get()->info()->strides_in_bytes()[channel_idx] / element_size_in_bytes; in prepare()