Searched refs:dim_val (Results 1 – 7 of 7) sorted by relevance
869 dim_val = tensor_shape.dimension_value(a.shape[dim])870 if dim_val is not None:871 return dim_val
260 int64 dim_val = Value(Dim(s, i)); in NumElements() local261 if (dim_val == kUnknownDim) { in NumElements()263 } else if (dim_val == 0) { in NumElements()266 size *= dim_val; in NumElements()
2132 auto dim_val = c->Value(dim); in SliceHelper() local2135 "Out of bounds slicing on dimension ", i, " of length ", dim_val, in SliceHelper()
2137 int64_t dim_val = (*dim_attr.begin()).getSExtValue(); in InferExpandDimsOpType() local2140 if (dim_val < -input_rank - 1 || dim_val > input_rank + 1) return unranked_ty; in InferExpandDimsOpType()2141 if (dim_val < 0) dim_val += input_rank + 1; in InferExpandDimsOpType()2144 shape.insert(shape.begin() + dim_val, 1); in InferExpandDimsOpType()
1487 int64_t dim_val = dim.value().getSExtValue(); in Verify() local1488 split_sizes.push_back(dim_val); in Verify()1489 if (dim_val == ShapedType::kDynamicSize) { in Verify()1496 total_dim_size += dim_val; in Verify()
3621 dim_val = dim3623 dim_val = tensor_util.constant_value(dim)3624 if dim_val is not None and not -shape.ndims <= dim_val < shape.ndims:3628 " dimensions in the input." % (dim_val, -shape.ndims, shape.ndims,
3025 int64_t dim_val = dim.value().getSExtValue(); in matchAndRewrite() local3026 split_sizes.push_back(dim_val); in matchAndRewrite()3027 if (dim_val == ShapedType::kDynamicSize) { in matchAndRewrite()3032 total_dim_size += dim_val; in matchAndRewrite()