Searched defs:b_indices (Results 1 – 5 of 5) sorted by relevance
32 const Tensor *backprop_val_grad, *a_indices, *b_indices, *sum_indices; in Compute() local
33 const Tensor *a_indices, *b_indices, *a_values_t, *b_values_t, *a_shape, in Compute() local
55 const auto b_indices = static_cast<int>(indices.shape()[3]); in max_unpooling_layer_internal() local
80 const auto b_indices = b_indices_tensor.matrix<int64_t>(); in Compute() local
56 ShapeHandle b_indices; in __anonad4a10b70202() local