Home
last modified time | relevance | path

Searched refs:reshaped_mean (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc1517 NodeDef reshaped_mean; in IsCpuCompatibleDataType() local
1518 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()