Searched defs:tensor_argmax (Results 1 – 1 of 1) sorted by relevance
126 Tensor<DenseIndex, 0, DataLayout> tensor_argmax; in test_simple_argmax() local170 Tensor<DenseIndex, 3, DataLayout> tensor_argmax; in test_argmax_dim() local