Searched refs:operand_dim_size (Results 1 – 3 of 3) sorted by relevance
296 Value operand_dim_size = in InsertDynamicMemrefCastOp() local300 operand_sizes[i] = operand_dim_size; in InsertDynamicMemrefCastOp()304 stride_so_far = b->create<MulIOp>(loc, stride_so_far, operand_dim_size); in InsertDynamicMemrefCastOp()
625 const int64_t operand_dim_size = operand_type.getDimSize(dim); in VerifyConcatenationOpTypes() local629 if (RankedTensorType::isDynamic(operand_dim_size) || in VerifyConcatenationOpTypes()633 result_dim_sizes[axis] += operand_dim_size; in VerifyConcatenationOpTypes()637 if (RankedTensorType::isDynamic(operand_dim_size)) continue; in VerifyConcatenationOpTypes()640 result_dim_sizes[dim] = operand_dim_size; in VerifyConcatenationOpTypes()645 if (result_dim_size != operand_dim_size) in VerifyConcatenationOpTypes()651 << operand_dim_size; in VerifyConcatenationOpTypes()
1822 auto operand_dim_size = operand.shape().dimensions(i); in HandleBroadcast() local1825 TF_RET_CHECK(operand_dim_size == broadcast_dim_size) << absl::StreamFormat( in HandleBroadcast()1828 i, broadcast->dimensions(i), operand_dim_size, broadcast_dim_size, in HandleBroadcast()