Searched refs:is_relu (Results 1 – 2 of 2) sorted by relevance
43 %is_relu = tfr.equal %act, %relu -> i144 %res = scf.if %is_relu -> !tfr.tensor {
352 bool is_relu = IsRelu(activation_node); in IsCpuCompatibleDataType() local354 return is_relu && is_spatial_conv && IsGpuCompatibleConv2D(&contraction_node); in IsCpuCompatibleDataType()