Searched refs:updates_dims (Results 1 – 2 of 2) sorted by relevance
104 std::vector<int64> 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()
1149 const int updates_dims = updates_shape.DimensionsCount(); in ScatterNd() local1153 for (int i = outer_dims; i < updates_dims; ++i) { in ScatterNd()