Searched defs:updates_dims (Results 1 – 3 of 3) sorted by relevance
104 std::vector<int64_t> updates_dims; in FuzzImpl() local
2579 absl::Span<const int64_t> updates_dims, in GetIterationSpaceImpl()2608 absl::Span<const int64_t> updates_dims, in IterationSpaceForUpdateScatterIndices()2618 absl::Span<const int64_t> updates_dims, in IterationSpaceForUpdateWindowIndices()2859 auto updates_dims = updates[0]->shape().dimensions(); in HandleScatter() local
671 const int updates_dims = updates_shape.DimensionsCount(); in ScatterNd() local