Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc1504 NodeDef reshaped_offset; in IsCpuCompatibleDataType() local
1505 reshaped_offset.set_name( in IsCpuCompatibleDataType()
1507 reshaped_offset.set_op("Reshape"); in IsCpuCompatibleDataType()
1508 reshaped_offset.set_device(fused_node.device()); in IsCpuCompatibleDataType()
1509 *reshaped_offset.add_input() = offset; in IsCpuCompatibleDataType()
1510 *reshaped_offset.add_input() = new_shape_name; in IsCpuCompatibleDataType()
1511 (*reshaped_offset.mutable_attr())["T"] = fused_node.attr().at("T"); in IsCpuCompatibleDataType()
1512 (*reshaped_offset.mutable_attr())["Tshape"].set_type(DT_INT32); in IsCpuCompatibleDataType()
1513 offset = reshaped_offset.name(); in IsCpuCompatibleDataType()
1514 mutation->AddNode(std::move(reshaped_offset), &status); in IsCpuCompatibleDataType()