Searched refs:result_dim_size (Results 1 – 3 of 3) sorted by relevance
133 int64 result_dim_size = dim_size - lhs_padding - rhs_padding; in DoMirrorPadGrad() local147 /*pad_hi=*/result_dim_size - lhs_padding - excluded_edges); in DoMirrorPadGrad()154 /*pad_lo=*/result_dim_size - rhs_padding - excluded_edges, in DoMirrorPadGrad()
318 Value result_dim_size = in InsertDynamicMemrefCastOp() local320 if (!result_dim_size.getType().isIndex()) { in InsertDynamicMemrefCastOp()321 result_dim_size = in InsertDynamicMemrefCastOp()322 b->create<IndexCastOp>(loc, result_dim_size, b->getIndexType()); in InsertDynamicMemrefCastOp()324 sizes.push_back(result_dim_size); in InsertDynamicMemrefCastOp()342 loc, CmpIPredicate::slt, operand_sizes[dim], result_dim_size); in InsertDynamicMemrefCastOp()
626 const int64_t result_dim_size = result_dim_sizes[dim]; in VerifyConcatenationOpTypes() local630 RankedTensorType::isDynamic(result_dim_size)) in VerifyConcatenationOpTypes()639 if (RankedTensorType::isDynamic(result_dim_size)) { in VerifyConcatenationOpTypes()645 if (result_dim_size != operand_dim_size) in VerifyConcatenationOpTypes()650 << FormatLoc(dim) << ", expected " << result_dim_size << ", got " in VerifyConcatenationOpTypes()