Searched defs:b_indices (Results 1 – 4 of 4) 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
55 ShapeHandle b_indices; in __anon9213d3ed0202() local