Searched refs:reshaped_offset (Results 1 – 1 of 1) sorted by relevance
1504 NodeDef reshaped_offset; in IsCpuCompatibleDataType() local1505 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()