Searched refs:contraction_node_def (Results 1 – 1 of 1) sorted by relevance
416 const auto* contraction_node_def = contraction_node_view->node(); in IsCpuCompatibleDataType() local419 bool is_contraction = IsConv2D(*contraction_node_def) || in IsCpuCompatibleDataType()420 IsMatMul(*contraction_node_def) || in IsCpuCompatibleDataType()421 IsDepthwiseConv2dNative(*contraction_node_def); in IsCpuCompatibleDataType()423 if (!is_contraction || !HaveSameDataType(node_def, contraction_node_def) || in IsCpuCompatibleDataType()426 IsInPreserveSet(ctx, contraction_node_def)) in IsCpuCompatibleDataType()469 const auto* contraction_node_def = contraction_node_view->node(); in IsCpuCompatibleDataType() local472 if (!IsMatMul(*contraction_node_def) && IsTanh(*node_def)) return false; in IsCpuCompatibleDataType()475 if (!(IsConv2D(*contraction_node_def) || IsMatMul(*contraction_node_def)) && in IsCpuCompatibleDataType()758 const auto* contraction_node_def = contraction_node_view->node(); in IsCpuCompatibleDataType() local[all …]