Searched refs:contraction_node (Results 1 – 1 of 1) sorted by relevance
332 const NodeDef& contraction_node = graph->node(matched.contraction); in IsCpuCompatibleDataType() local333 if (!IsConv2D(contraction_node)) return false; in IsCpuCompatibleDataType()336 ctx.graph_properties.GetInputProperties(contraction_node.name()); in IsCpuCompatibleDataType()354 return is_relu && is_spatial_conv && IsGpuCompatibleConv2D(&contraction_node); in IsCpuCompatibleDataType()1297 NodeDef contraction_node; in IsCpuCompatibleDataType() local1299 contraction_node.set_name(add.name()); in IsCpuCompatibleDataType()1300 contraction_node.set_device(contraction.device()); in IsCpuCompatibleDataType()1301 contraction_node.add_input( in IsCpuCompatibleDataType()1303 contraction_node.add_input( in IsCpuCompatibleDataType()1305 contraction_node.add_input(bias_add.input(1)); // 2: bias in IsCpuCompatibleDataType()[all …]