Searched refs:begin_dim (Results 1 – 3 of 3) sorted by relevance
446 int64 begin_dim = shape_spec.processing_to_sparse_mapping[i]; in CompileAsDynamicUpdateSlice() local455 xla::Slice(ctx->Input(1), {begin_dim}, {begin_dim + 1}, {1}); in CompileAsDynamicUpdateSlice()
3305 begin_dim = math_ops.reduce_sum(d[:self.replica_id_in_sync_group])3306 end_dim = begin_dim + array_ops.shape(xs[i])[axis]3308 grad, axis=axis, indices=math_ops.range(begin_dim, end_dim))
639 const int begin_dim = begin_shape.size(); in RewriteEllipsisMask() local641 if (begin_dim != 1) return failure(); in RewriteEllipsisMask()