Searched refs:DimensionVector (Results 1 – 4 of 4) sorted by relevance
178 DimensionVector window_index(rank); in IterateThroughWindow()946 DimensionVector lhs_index(lhs_rank); in HandleConvolution()947 DimensionVector rhs_index(rhs_rank); in HandleConvolution()948 DimensionVector rhs_spatial_index(dnums.kernel_spatial_dimensions_size()); in HandleConvolution()1084 DimensionVector lhs_index(lhs_rank); in HandleDot()1085 DimensionVector rhs_index(rhs_rank); in HandleDot()1558 DimensionVector source_index(rank); in HandleSelectAndScatter()1665 DimensionVector window_index(window.dimensions_size()); in HandleReduceWindow()1666 DimensionVector operand_index(ShapeUtil::Rank(operand_literal.shape())); in HandleReduceWindow()1722 DimensionVector operand_index(rank); in HandleSlice()[all …]
644 DimensionVector base;645 DimensionVector step;1262 DimensionVector minor_scan_indexes(rank, 0); in Populate()1307 DimensionVector bounds = {times}; in Replicate()1319 DimensionVector output_indices(bounds.size(), 0); in Replicate()
221 DimensionVector src_indexes(src_base.size(), 0); in CopySliceFromInternal()222 DimensionVector dest_indexes(dest_base.size(), 0); in CopySliceFromInternal()808 DimensionVector result_dimensions; in Slice()822 DimensionVector new_indices(ShapeUtil::Rank(result_shape)); in Slice()
55 using DimensionVector = tensorflow::gtl::InlinedVector<int64, kInlineRank>; variable