also prune out dependencies of fed tensors. Remove special handling of control dependencies from the introduced placeholders. For a compilation going to XLA, those control dependencies would not really be doing anything anyway (and the current code was only handling Placeholder, not PlaceholderV2). PiperOrigin-RevId: 161157126
57 lines
894 B
Plaintext
57 lines
894 B
Plaintext
node {
|
|
name : "x_const"
|
|
op : "Const"
|
|
attr {
|
|
key: "value"
|
|
value {
|
|
tensor { dtype: DT_INT32 tensor_shape { dim { size: 1 } } int_val: 1 }
|
|
}
|
|
}
|
|
attr {
|
|
key : "dtype"
|
|
value {
|
|
type : DT_INT32
|
|
}
|
|
}
|
|
}
|
|
node {
|
|
name : "y_const"
|
|
op : "Const"
|
|
attr {
|
|
key: "value"
|
|
value {
|
|
tensor { dtype: DT_INT32 tensor_shape { dim { size: 1 } } int_val: 2 }
|
|
}
|
|
}
|
|
attr {
|
|
key: "dtype"
|
|
value {
|
|
type: DT_INT32
|
|
}
|
|
}
|
|
}
|
|
node {
|
|
name : "y_reshape"
|
|
op : "Reshape"
|
|
input : "y_const"
|
|
input : "y_shape"
|
|
attr { key: "T" value { type: DT_INT32 } }
|
|
# Attribute TShape not specified; needs to be set to its default
|
|
# by tfcompile.
|
|
}
|
|
node {
|
|
name : "x_y_sum"
|
|
op : "Add"
|
|
input : "x_const"
|
|
input : "y_reshape"
|
|
attr {
|
|
key : "T"
|
|
value {
|
|
type: DT_INT32
|
|
}
|
|
}
|
|
}
|
|
versions {
|
|
producer: 15
|
|
}
|