Searched refs:updates_dims (Results 1 – 3 of 3) sorted by relevance
104 std::vector<int64_t> updates_dims; in FuzzImpl() local106 updates_dims.push_back(indices_dims[i]); in FuzzImpl()111 updates_dims.push_back(flat_shape(i)); in FuzzImpl()114 Tensor updates_tensor(tensorflow::DT_INT32, TensorShape(updates_dims)); in FuzzImpl()
2579 absl::Span<const int64_t> updates_dims, in GetIterationSpaceImpl() argument2581 int64_t updates_rank = updates_dims.size(); in GetIterationSpaceImpl()2590 index_count[i] = updates_dims[i]; in GetIterationSpaceImpl()2596 index_count[i] = updates_dims[i]; in GetIterationSpaceImpl()2608 absl::Span<const int64_t> updates_dims, in IterationSpaceForUpdateScatterIndices() argument2610 return GetIterationSpaceImpl</*kForUpdateWindowIndices=*/false>(updates_dims, in IterationSpaceForUpdateScatterIndices()2618 absl::Span<const int64_t> updates_dims, in IterationSpaceForUpdateWindowIndices() argument2620 return GetIterationSpaceImpl</*kForUpdateWindowIndices=*/true>(updates_dims, in IterationSpaceForUpdateWindowIndices()2859 auto updates_dims = updates[0]->shape().dimensions(); in HandleScatter() local2863 IterationSpaceForUpdateScatterIndices(updates_dims, dim_numbers); in HandleScatter()[all …]
671 const int updates_dims = updates_shape.DimensionsCount(); in ScatterNd() local675 for (int i = outer_dims; i < updates_dims; ++i) { in ScatterNd()