Searched defs:updates_rank (Results 1 – 7 of 7) sorted by relevance
36 int updates_rank, const Mesh& mesh) { in GetOutputLayout()88 const int updates_rank = ValueRank(scatter_op.updates()); in TensorScatterOpExpand() local160 const int updates_rank = ValueRank(scatter_op.updates()); in TensorScatterOpComputeLayoutForward() local189 const int updates_rank = ValueRank(scatter_op.updates()); in TensorScatterOpComputeLayoutBackward() local
143 int64_t updates_rank = updates_shape.rank(); in XlaScatter() local
43 int64_t updates_rank = updates->shape().rank(); in FlattenAndTransposeUpdates() local
100 const int64_t updates_rank = updates->shape().rank(); in PermuteScatterAndWindowDims() local
2581 int64_t updates_rank = updates_dims.size(); in GetIterationSpaceImpl() local2638 int64_t updates_rank, const Literal* scatter_indices) in UpdateScatterIndexToInputIndex()
3578 const int64_t updates_rank = updates_shape.rank(); in ValidateScatterDimensionNumbers() local
4512 int64_t updates_rank = in matchAndRewrite() local