Searched refs:a_batch_idx (Results 1 – 2 of 2) sorted by relevance
218 const int a_batch_idx = bcast.IsBroadcastingRequired() in Run() local225 matmul_op->inputs = {slice_a_outputs[a_batch_idx], in Run()
477 const size_t a_batch_idx = _gemm_info.reinterpret_input_as_3d != 0 ? 3 : 2; in run() local478 const size_t a_multi_idx = a_batch_idx + 1; in run()482 … int batch_stride_a = a->info()->strides_in_bytes()[a_batch_idx] / a->info()->element_size(); in run()