Searched refs:stride_op (Results 1 – 2 of 2) sorted by relevance
74 auto stride_op = ops::Const(root, {stride}); in TestStridedSlice() local75 auto slice = ops::StridedSlice(root, input, begin_op, end_op, stride_op, in TestStridedSlice()89 TF_ASSERT_OK(m.AddNode(stride_op.node())); in TestStridedSlice()
710 auto stride_op = in RewriteEllipsisMask() local715 end_op.getResult(), stride_op.getResult(), in RewriteEllipsisMask()