Searched refs:reshaped_mean (Results 1 – 1 of 1) sorted by relevance
1517 NodeDef reshaped_mean; in IsCpuCompatibleDataType() local1518 reshaped_mean.set_name( in IsCpuCompatibleDataType()1520 reshaped_mean.set_op("Reshape"); in IsCpuCompatibleDataType()1521 reshaped_mean.set_device(fused_node.device()); in IsCpuCompatibleDataType()1522 *reshaped_mean.add_input() = mean; in IsCpuCompatibleDataType()1523 *reshaped_mean.add_input() = new_shape_name; in IsCpuCompatibleDataType()1524 (*reshaped_mean.mutable_attr())["T"] = fused_node.attr().at("T"); in IsCpuCompatibleDataType()1525 (*reshaped_mean.mutable_attr())["Tshape"].set_type(DT_INT32); in IsCpuCompatibleDataType()1526 mean = reshaped_mean.name(); in IsCpuCompatibleDataType()1527 mutation->AddNode(std::move(reshaped_mean), &status); in IsCpuCompatibleDataType()