Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/runtime/
Dgraph_executor.cpp292 if (input_tensor_lists_.count(output_index) != 0) { in apply()
293 size_t list_size = input_tensor_lists_[output_index]; in apply()
332 input_tensor_lists_.insert({index_, value.toTensorList().size()}); in addOutputForIValue()
408 std::map<size_t, size_t> input_tensor_lists_; member