Searched defs:idx_tensor (Results 1 – 4 of 4) sorted by relevance
31 int idx_tensor = 0; in _foreach_tensor() local365 [&](size_t idx_tensor, size_t _, const at::Tensor& t) { in autogradNotImplementedFallbackImpl()381 [&](size_t idx_tensor, size_t idx_ret, const at::Tensor& t) { in autogradNotImplementedFallbackImpl()458 [&](size_t idx_tensor, size_t idx_ret, const at::Tensor& t) { in autogradNotImplementedFallbackImpl()
398 Tensor* idx_tensor; in ReturnIdxTensor() local
4149 const TensorId idx_tensor = ParseTensorName(reduction_node->input(1)); in TrySimplify() local
10730 def idx_tensor(size, max_val): function