Searched refs:reshaped_variance (Results 1 – 1 of 1) sorted by relevance
1530 NodeDef reshaped_variance; in IsCpuCompatibleDataType() local1531 reshaped_variance.set_name( in IsCpuCompatibleDataType()1533 reshaped_variance.set_op("Reshape"); in IsCpuCompatibleDataType()1534 reshaped_variance.set_device(fused_node.device()); in IsCpuCompatibleDataType()1535 *reshaped_variance.add_input() = variance; in IsCpuCompatibleDataType()1536 *reshaped_variance.add_input() = new_shape_name; in IsCpuCompatibleDataType()1537 (*reshaped_variance.mutable_attr())["T"] = fused_node.attr().at("T"); in IsCpuCompatibleDataType()1538 (*reshaped_variance.mutable_attr())["Tshape"].set_type(DT_INT32); in IsCpuCompatibleDataType()1539 variance = reshaped_variance.name(); in IsCpuCompatibleDataType()1540 mutation->AddNode(std::move(reshaped_variance), &status); in IsCpuCompatibleDataType()