Searched refs:reverse_index_map_t (Results 1 – 2 of 2) sorted by relevance
12 context, TensorShapeUtils::IsVector(reverse_index_map_t->shape()),14 reverse_index_map_t->shape().DebugString()));16 const auto reverse_index_map = reverse_index_map_t->vec<int64>();20 Although `reverse_index_map_t` and `grad_values_t` are accessed in a similar21 pattern, only `reverse_index_map_t` is validated to be of proper shape. Hence,
68 Tensor* reverse_index_map_t = nullptr; in operator ()() local70 kReverseIndexMapOutput, TensorShape({N}), &reverse_index_map_t)); in operator ()()71 reverse_index_map = reverse_index_map_t->vec<Tindex>().data(); in operator ()()295 const Tensor* reverse_index_map_t; in Compute() local298 context->input("reverse_index_map", &reverse_index_map_t)); in Compute()302 context, TensorShapeUtils::IsVector(reverse_index_map_t->shape()), in Compute()304 reverse_index_map_t->shape().DebugString())); in Compute()309 const auto reverse_index_map = reverse_index_map_t->vec<Tindex>(); in Compute()312 const Tindex N = reverse_index_map_t->shape().dim_size(0); in Compute()