Searched refs:graph_io_tensors (Results 1 – 1 of 1) sorted by relevance
1611 std::vector<torch::jit::Value*> graph_io_tensors; in deduceMemoryLayoutPolicy() local1615 std::back_inserter(graph_io_tensors), in deduceMemoryLayoutPolicy()1620 std::back_inserter(graph_io_tensors), in deduceMemoryLayoutPolicy()1625 auto prefer_channels_last = (!graph_io_tensors.empty()); in deduceMemoryLayoutPolicy()1626 for (auto el : graph_io_tensors) { in deduceMemoryLayoutPolicy()