Searched defs:num_returns (Results 1 – 13 of 13) sorted by relevance
105 const auto num_returns = schema.returns().size(); in basicAutogradNotImplementedFallbackImpl() local241 const auto num_returns = schema.returns().size(); in autogradNotImplementedFallbackImpl() local505 const auto num_returns = schema.returns().size(); in autogradNotImplementedInplaceOrViewFallbackImpl() local
160 const auto num_returns = schema_returns.size(); in pushPyOutToStack() local
261 const auto num_returns = schema.returns().size(); in batchedTensorForLoopFallback() local401 const auto num_returns = schema.returns().size(); in batchedNestedTensorForLoopFallback() local
108 const auto num_returns = schema.returns().size(); in boxed_reduction_batch_rule() local
51 auto num_returns = op.schema().returns().size(); in generic_wrapper_fallback() local
213 const auto& num_returns = schema_returns.size(); in cpu_fallback() local
252 const auto num_returns = schema.returns().size(); in batchedTensorForLoopFallback() local
100 const auto num_returns = schema.returns().size(); in functionalizeFallback() local
875 const auto num_returns = schema.returns().size(); in functionalize_op_helper() local
287 const auto& num_returns = schema_returns.size(); in ts_eager_fallback() local
489 int num_returns = 0; in si_llvm_build_ps_prolog() local
346 unsigned i, num_returns, num_return_sgprs; in si_init_shader_args() local
1832 int64_t num_returns = remote_rets->size(); in RunInternal() local