Searched refs:contraction_node (Results 1 – 1 of 1) sorted by relevance
472 const NodeDef& contraction_node = graph->node(matched.contraction); in IsGpuCompatible() local473 if (IsConv2D(contraction_node)) { in IsGpuCompatible()475 ctx.graph_properties.GetInputProperties(contraction_node.name()); in IsGpuCompatible()506 return is_supported_conv && IsGpuCompatibleConv2D(ctx, &contraction_node); in IsGpuCompatible()507 } else if (IsMatMul(contraction_node)) { in IsGpuCompatible()508 return IsGpuCompatibleMatMul(ctx, &contraction_node); in IsGpuCompatible()529 const NodeDef& contraction_node = graph->node(matched.contraction); in IsGpuCompatible() local530 if (!IsMatMul(contraction_node)) return false; in IsGpuCompatible()532 return IsGpuCompatibleMatMul(ctx, &contraction_node); in IsGpuCompatible()2454 NodeDef contraction_node; in AddFusedContractionNode() local[all …]