Searched defs:new_index_shape (Results 1 – 2 of 2) sorted by relevance
109 TensorShape new_index_shape({num_outputs}); in Compute() local
833 VmapDimVector new_index_shape(self_size.size(), 1); in get_expanded_index() local839 VmapDimVector new_index_shape = {self_size.begin(), self_size.end()}; in get_expanded_index() local