Searched refs:reshaped_scale (Results 1 – 1 of 1) sorted by relevance
1491 NodeDef reshaped_scale; in IsCpuCompatibleDataType() local1492 reshaped_scale.set_name( in IsCpuCompatibleDataType()1494 reshaped_scale.set_op("Reshape"); in IsCpuCompatibleDataType()1495 reshaped_scale.set_device(fused_node.device()); in IsCpuCompatibleDataType()1496 *reshaped_scale.add_input() = scale; in IsCpuCompatibleDataType()1497 *reshaped_scale.add_input() = new_shape_name; in IsCpuCompatibleDataType()1498 (*reshaped_scale.mutable_attr())["T"] = fused_node.attr().at("T"); in IsCpuCompatibleDataType()1499 (*reshaped_scale.mutable_attr())["Tshape"].set_type(DT_INT32); in IsCpuCompatibleDataType()1500 scale = reshaped_scale.name(); in IsCpuCompatibleDataType()1501 mutation->AddNode(std::move(reshaped_scale), &status); in IsCpuCompatibleDataType()