Searched refs:NodeIsOnCpu (Results 1 – 4 of 4) sorted by relevance
351 return NodeIsOnCpu(conv2d) && IsCpuCompatibleDataType(conv2d) && in IsCpuCompatibleConv2D()357 return NodeIsOnCpu(conv3d) && IsCpuCompatibleDataType(conv3d) && in IsCpuCompatibleConv3D()375 return NodeIsOnCpu(matmul) && IsCpuCompatibleDataType(matmul); in IsCpuCompatibleMatMul()381 return NodeIsOnCpu(dw_conv2d) && IsCpuCompatibleDataType(dw_conv2d); in IsCpuCompatibleDepthwiseConv2dNative()583 if (!NodeIsOnCpu(node_def)) return false; in IsBiasSemanticAdd()847 if (!IsConv2D(*conv2d_node_def) || !NodeIsOnCpu(conv2d_node_def) || in FindConv2DWithBatchNorm()946 if (!NodeIsOnCpu(node_def)) return false; in FindPadWithConv3D()985 if (!NodeIsOnCpu(node_def)) return false; in FindContractionWithBiasAddAndAdd()1032 if (!NodeIsOnCpu(node_def)) return false; in FindContractionWithBiasAndAddActivation()1347 if (!NodeIsOnCpu(max_node_def) && !IsMKLEnabled()) return false; in FindMulAndMaximum()[all …]
197 bool NodeIsOnCpu(const NodeDef& node) { in NodeIsOnCpu() function2542 if (!is_complex || NodeIsOnCpu(*node)) { in TrySimplify()3204 if (NodeIsOnCpu(*node)) { in TrySimplify()3680 if (!NodeIsOnCpu(node)) { in CanOptimize()
336 bool NodeIsOnCpu(const NodeDef* node);
139 bool NodeIsOnCpu(const NodeDef* node) { in NodeIsOnCpu() function