Searched refs:result_batch_shape (Results 1 – 4 of 4) sorted by relevance
297 const auto& result_batch_shape = bcast.output_batch_shape().dim_sizes(); in matchAndRewrite() local298 std::vector<int64_t> result_shape(result_batch_shape.begin(), in matchAndRewrite()299 result_batch_shape.end()); in matchAndRewrite()
244 const auto& result_batch_shape = bcast.output_batch_shape().dim_sizes(); in Run() local247 std::transform(result_batch_shape.begin(), result_batch_shape.end(), in Run()
184 llvm::SmallVector<int64_t, 4> result_batch_shape; in Verify() local207 result_batch_shape)) in Verify()224 for (int i = 0; i < result_batch_shape.size(); ++i) { in Verify()226 output_shape[i] != result_batch_shape[i]) in Verify()229 << result_batch_shape[i] << " and output batch dimension " in Verify()
2790 auto result_batch_shape = rewriter->create<shape::BroadcastOp>( in BroadcastBatchMatMulV2Operands() local2804 rewriter->create<shape::ConcatOp>(loc, result_batch_shape, tail_shape); in BroadcastBatchMatMulV2Operands()