Searched refs:activation_node (Results 1 – 2 of 2) sorted by relevance
419 Node* activation_node; in MaybeFuseActivation() local421 NewPassthroughNode(graph, node, outputs[0], &activation_node)); in MaybeFuseActivation()422 activation_node->operation.type = ToString(OperationType::RELU); in MaybeFuseActivation()423 activation_node->operation.attributes = attr; in MaybeFuseActivation()427 Node* activation_node; in MaybeFuseActivation() local429 NewPassthroughNode(graph, node, outputs[0], &activation_node)); in MaybeFuseActivation()430 activation_node->operation.type = ToString(OperationType::TANH); in MaybeFuseActivation()434 Node* activation_node; in MaybeFuseActivation() local436 NewPassthroughNode(graph, node, outputs[0], &activation_node)); in MaybeFuseActivation()437 activation_node->operation.type = ToString(OperationType::SIGMOID); in MaybeFuseActivation()
351 const NodeDef& activation_node = graph->node(matched.activation); in IsCpuCompatibleDataType() local352 bool is_relu = IsRelu(activation_node); in IsCpuCompatibleDataType()