Searched refs:num_outer_dims (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1557 const int64 num_outer_dims = c->Rank(indices_shape) - 1; in ScatterNdUpdateShape() local
1567 c->Subshape(indices_shape, 0, num_outer_dims, &prefix_indices)); in ScatterNdUpdateShape()
1570 c->Subshape(updates_shape, 0, num_outer_dims, &prefix_updates)); in ScatterNdUpdateShape()
1575 "The outer ", num_outer_dims, in ScatterNdUpdateShape()
1577 " must match the outer ", num_outer_dims, in ScatterNdUpdateShape()
1586 c->Subshape(updates_shape, num_outer_dims, &suffix_updates)); in ScatterNdUpdateShape()
1592 " must match the inner ", c->Rank(updates_shape) - num_outer_dims, in ScatterNdUpdateShape()
|