Searched refs:mkl_node_output_slot (Results 1 – 1 of 1) sorted by relevance
1932 int* mkl_node_output_slot);2170 int mkl_node_output_slot = 0; in GetNodesProducingMklTensorList() local2172 &mkl_node_output_slot); in GetNodesProducingMklTensorList()2174 NodeBuilder::NodeOut(mkl_node, mkl_node_output_slot)); in GetNodesProducingMklTensorList()2186 int n_output_slot, Node** mkl_node, int* mkl_node_output_slot) { in GetNodeProducingMklTensor() argument2189 DCHECK(mkl_node_output_slot); in GetNodeProducingMklTensor()2200 *mkl_node_output_slot = in GetNodeProducingMklTensor()2209 *mkl_node_output_slot = 0; in GetNodeProducingMklTensor()2328 int mkl_node_output_slot = 0; in SetUpContiguousInputs() local2334 &mkl_node_output_slot); in SetUpContiguousInputs()[all …]