Searched defs:cast_node (Results 1 – 6 of 6) sorted by relevance
18 Node* cast_node = graph->create(c10::onnx::Cast); in CreateCastToBoolNode() local35 Node* cast_node = CreateCastToBoolNode(cond_val, graph); in InsertCastForCond() local300 auto* cast_node = InsertCastForCond(cond_val, graph, node, opset_version); in FixupONNXLoopNodeInputs() local316 auto* cast_node = InsertCastForCond( in FixupONNXLoopNodeInputs() local471 Node* cast_node = in ONNXFixupUninitializedOutput() local
339 Node* cast_node = n->owningGraph()->create(onnx::Cast); in UpdateScalarTypeForInputs() local369 Node* cast_node = n->owningGraph()->create(onnx::Cast, 1); in RecoverScalarTypeForOutput() local
55 Node* cast_node = graph->create(onnx::Cast, 1); in CastAllConstantToFloating() local
924 Node* cast_node = b->owningGraph()->create(onnx::Cast, 1); in fuseLogSoftmaxNllLoss() local
1664 const NodeDef* cast_node = nullptr; in TEST_F() local1716 const NodeDef* cast_node = nullptr; in TEST_F() local2175 const NodeDef* cast_node = node_map.GetNode(optimized_cast_name); in TEST_F() local
4261 NodeDef* cast_node = nullptr; in TrySimplify() local