Searched refs:operand_dim_size (Results 1 – 2 of 2) sorted by relevance
774 const int64_t operand_dim_size = operand_type.getDimSize(dim); in VerifyConcatenationOpTypes() local778 if (ShapedType::isDynamic(operand_dim_size) || in VerifyConcatenationOpTypes()782 result_dim_sizes[axis] += operand_dim_size; in VerifyConcatenationOpTypes()787 if (ShapedType::isDynamic(operand_dim_size)) continue; in VerifyConcatenationOpTypes()790 result_dim_sizes[dim] = operand_dim_size; in VerifyConcatenationOpTypes()795 if (result_dim_size != operand_dim_size) in VerifyConcatenationOpTypes()801 << operand_dim_size; in VerifyConcatenationOpTypes()
2977 auto operand_dim_size = operand.shape().dimensions(i); in HandleBroadcast() local2980 TF_RET_CHECK(operand_dim_size == broadcast_dim_size) << absl::StreamFormat( in HandleBroadcast()2983 i, broadcast->dimensions(i), operand_dim_size, broadcast_dim_size, in HandleBroadcast()