Home
last modified time | relevance | path

Searched refs:df_stack (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/test/cpp/jit/
Dtest_utils.cpp52 Stack df_stack; in runGradient() local
53 df_stack.insert( in runGradient()
54 df_stack.end(), tensor_grads_in.begin(), tensor_grads_in.end()); in runGradient()
56 df_stack.push_back(tensors_in[offset]); in runGradient()
58 df_stack.push_back(f_stack[offset]); in runGradient()
59 df_interpreter.run(df_stack); in runGradient()
60 unpackReturnTuple(df_stack); in runGradient()
63 return std::make_pair(as_tensorlist(f_stack), as_tensorlist(df_stack)); in runGradient()