diff --git a/source/tests/infer/sea_012.pbtxt b/source/tests/infer/sea_012.pbtxt new file mode 100644 index 0000000000..4be031a4c7 --- /dev/null +++ b/source/tests/infer/sea_012.pbtxt @@ -0,0 +1,13409 @@ +node { + name: "t_avg" + op: "Const" + attr { + key: "dtype" + value { + type: DT_DOUBLE + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_DOUBLE + tensor_shape { + dim { + size: 2 + } + dim { + size: 8 + } + } + tensor_content: "=jt\221\306\201\346?\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000=jt\221\306\201\346?\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\370\005&\t\377\265\352?\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\370\005&\t\377\265\352?\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000" + } + } + } +} +node { + name: "t_avg/read" + op: "Identity" + input: "t_avg" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@t_avg" + } + } + } +} +node { + name: "t_std" + op: "Const" + attr { + key: "dtype" + value { + type: DT_DOUBLE + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_DOUBLE + tensor_shape { + dim { + size: 2 + } + dim { + size: 8 + } + } + tensor_content: "\007~\223\256\265w\325?;e!U\261\307\334?;e!U\261\307\334?;e!U\261\307\334?\007~\223\256\265w\325?;e!U\261\307\334?;e!U\261\307\334?;e!U\261\307\334?\207\031\275\035!?\310?UH\363[\260\234\337?UH\363[\260\234\337?UH\363[\260\234\337?\207\031\275\035!?\310?UH\363[\260\234\337?UH\363[\260\234\337?UH\363[\260\234\337?" + } + } + } +} +node { + name: "t_std/read" + op: "Identity" + input: "t_std" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@t_std" + } + } + } +} +node { + name: "t_rcut" + op: "Const" + attr { + key: "dtype" + value { + type: DT_DOUBLE + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_DOUBLE + tensor_shape { + } + double_val: 6.0 + } + } + } +} +node { + name: "t_ntypes" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 2 + } + } + } +} +node { + name: "t_coord" + op: "Placeholder" + attr { + key: "dtype" + value { + type: DT_DOUBLE + } + } + attr { + key: "shape" + value { + shape { + dim { + size: -1 + } + } + } + } +} +node { + name: "t_type" + op: "Placeholder" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "shape" + value { + shape { + dim { + size: -1 + } + } + } + } +} +node { + name: "t_natoms" + op: "Placeholder" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "shape" + value { + shape { + dim { + size: 4 + } + } + } + } +} +node { + name: "t_box" + op: "Placeholder" + attr { + key: "dtype" + value { + type: DT_DOUBLE + } + } + attr { + key: "shape" + value { + shape { + dim { + size: -1 + } + dim { + size: 9 + } + } + } + } +} +node { + name: "t_mesh" + op: "Placeholder" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "shape" + value { + shape { + dim { + size: -1 + } + } + } + } +} +node { + name: "strided_slice/stack" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "strided_slice/stack_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 2 + } + } + } +} +node { + name: "strided_slice/stack_2" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "strided_slice" + op: "StridedSlice" + input: "t_natoms" + input: "strided_slice/stack" + input: "strided_slice/stack_1" + input: "strided_slice/stack_2" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "begin_mask" + value { + i: 0 + } + } + attr { + key: "ellipsis_mask" + value { + i: 0 + } + } + attr { + key: "end_mask" + value { + i: 0 + } + } + attr { + key: "new_axis_mask" + value { + i: 0 + } + } + attr { + key: "shrink_axis_mask" + value { + i: 1 + } + } +} +node { + name: "mul/y" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 3 + } + } + } +} +node { + name: "mul" + op: "Mul" + input: "strided_slice" + input: "mul/y" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "Reshape/shape/0" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: -1 + } + } + } +} +node { + name: "Reshape/shape" + op: "Pack" + input: "Reshape/shape/0" + input: "mul" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "axis" + value { + i: 0 + } + } +} +node { + name: "Reshape" + op: "Reshape" + input: "t_coord" + input: "Reshape/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "strided_slice_1/stack" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "strided_slice_1/stack_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 2 + } + } + } +} +node { + name: "strided_slice_1/stack_2" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "strided_slice_1" + op: "StridedSlice" + input: "t_natoms" + input: "strided_slice_1/stack" + input: "strided_slice_1/stack_1" + input: "strided_slice_1/stack_2" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "begin_mask" + value { + i: 0 + } + } + attr { + key: "ellipsis_mask" + value { + i: 0 + } + } + attr { + key: "end_mask" + value { + i: 0 + } + } + attr { + key: "new_axis_mask" + value { + i: 0 + } + } + attr { + key: "shrink_axis_mask" + value { + i: 1 + } + } +} +node { + name: "Reshape_1/shape/0" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: -1 + } + } + } +} +node { + name: "Reshape_1/shape" + op: "Pack" + input: "Reshape_1/shape/0" + input: "strided_slice_1" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "axis" + value { + i: 0 + } + } +} +node { + name: "Reshape_1" + op: "Reshape" + input: "t_type" + input: "Reshape_1/shape" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "DescrptNorot" + op: "DescrptNorot" + input: "Reshape" + input: "Reshape_1" + input: "t_natoms" + input: "t_box" + input: "t_mesh" + input: "t_avg/read" + input: "t_std/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "rcut_a" + value { + f: -1.0 + } + } + attr { + key: "rcut_r" + value { + f: 6.0 + } + } + attr { + key: "rcut_r_smth" + value { + f: 5.800000190734863 + } + } + attr { + key: "sel_a" + value { + list { + i: 1 + i: 1 + } + } + } + attr { + key: "sel_r" + value { + list { + i: 0 + i: 0 + } + } + } +} +node { + name: "Reshape_2/shape" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\377\377\377\377\010\000\000\000" + } + } + } +} +node { + name: "Reshape_2" + op: "Reshape" + input: "DescrptNorot" + input: "Reshape_2/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "strided_slice_2/stack" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 0 + } + } + } +} +node { + name: "strided_slice_2/stack_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "strided_slice_2/stack_2" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "strided_slice_2" + op: "StridedSlice" + input: "t_natoms" + input: "strided_slice_2/stack" + input: "strided_slice_2/stack_1" + input: "strided_slice_2/stack_2" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "begin_mask" + value { + i: 0 + } + } + attr { + key: "ellipsis_mask" + value { + i: 0 + } + } + attr { + key: "end_mask" + value { + i: 0 + } + } + attr { + key: "new_axis_mask" + value { + i: 0 + } + } + attr { + key: "shrink_axis_mask" + value { + i: 1 + } + } +} +node { + name: "mul_1/x" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 8 + } + } + } +} +node { + name: "mul_1" + op: "Mul" + input: "mul_1/x" + input: "strided_slice_2" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "Reshape_3/shape/0" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: -1 + } + } + } +} +node { + name: "Reshape_3/shape" + op: "Pack" + input: "Reshape_3/shape/0" + input: "mul_1" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "axis" + value { + i: 0 + } + } +} +node { + name: "Reshape_3" + op: "Reshape" + input: "Reshape_2" + input: "Reshape_3/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "strided_slice_3/stack" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 2 + } + } + } +} +node { + name: "strided_slice_3/stack_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 3 + } + } + } +} +node { + name: "strided_slice_3/stack_2" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "strided_slice_3" + op: "StridedSlice" + input: "t_natoms" + input: "strided_slice_3/stack" + input: "strided_slice_3/stack_1" + input: "strided_slice_3/stack_2" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "begin_mask" + value { + i: 0 + } + } + attr { + key: "ellipsis_mask" + value { + i: 0 + } + } + attr { + key: "end_mask" + value { + i: 0 + } + } + attr { + key: "new_axis_mask" + value { + i: 0 + } + } + attr { + key: "shrink_axis_mask" + value { + i: 1 + } + } +} +node { + name: "mul_2/y" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 8 + } + } + } +} +node { + name: "mul_2" + op: "Mul" + input: "strided_slice_3" + input: "mul_2/y" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "Slice/begin" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\000\000\000\000\000\000\000\000" + } + } + } +} +node { + name: "Slice/size/0" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: -1 + } + } + } +} +node { + name: "Slice/size" + op: "Pack" + input: "Slice/size/0" + input: "mul_2" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "axis" + value { + i: 0 + } + } +} +node { + name: "Slice" + op: "Slice" + input: "Reshape_3" + input: "Slice/begin" + input: "Slice/size" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "Reshape_4/shape" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\377\377\377\377\010\000\000\000" + } + } + } +} +node { + name: "Reshape_4" + op: "Reshape" + input: "Slice" + input: "Reshape_4/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "strided_slice_4/stack" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 2 + } + } + } +} +node { + name: "strided_slice_4/stack_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 3 + } + } + } +} +node { + name: "strided_slice_4/stack_2" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "strided_slice_4" + op: "StridedSlice" + input: "t_natoms" + input: "strided_slice_4/stack" + input: "strided_slice_4/stack_1" + input: "strided_slice_4/stack_2" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "begin_mask" + value { + i: 0 + } + } + attr { + key: "ellipsis_mask" + value { + i: 0 + } + } + attr { + key: "end_mask" + value { + i: 0 + } + } + attr { + key: "new_axis_mask" + value { + i: 0 + } + } + attr { + key: "shrink_axis_mask" + value { + i: 1 + } + } +} +node { + name: "add/x" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 0 + } + } + } +} +node { + name: "add" + op: "Add" + input: "add/x" + input: "strided_slice_4" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "DS_layer_type_0/Slice/begin" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\000\000\000\000\000\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_0/Slice/size" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\377\377\377\377\004\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_0/Slice" + op: "Slice" + input: "Reshape_4" + input: "DS_layer_type_0/Slice/begin" + input: "DS_layer_type_0/Slice/size" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "DS_layer_type_0/Reshape/shape" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\377\377\377\377\004\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_0/Reshape" + op: "Reshape" + input: "DS_layer_type_0/Slice" + input: "DS_layer_type_0/Reshape/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "DS_layer_type_0/Slice_1/begin" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\000\000\000\000\000\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_0/Slice_1/size" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\377\377\377\377\001\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_0/Slice_1" + op: "Slice" + input: "DS_layer_type_0/Reshape" + input: "DS_layer_type_0/Slice_1/begin" + input: "DS_layer_type_0/Slice_1/size" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "DS_layer_type_0/Reshape_1/shape" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\377\377\377\377\001\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_0/Reshape_1" + op: "Reshape" + input: "DS_layer_type_0/Slice_1" + input: "DS_layer_type_0/Reshape_1/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "DS_layer_type_0/matrix_1_0" + op: "Const" + attr { + key: "dtype" + value { + type: DT_DOUBLE + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_DOUBLE + tensor_shape { + dim { + size: 1 + } + dim { + size: 1 + } + } + double_val: -0.5929978920085608 + } + } + } +} +node { + name: "DS_layer_type_0/matrix_1_0/read" + op: "Identity" + input: "DS_layer_type_0/matrix_1_0" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@DS_layer_type_0/matrix_1_0" + } + } + } +} +node { + name: "DS_layer_type_0/bias_1_0" + op: "Const" + attr { + key: "dtype" + value { + type: DT_DOUBLE + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_DOUBLE + tensor_shape { + dim { + size: 1 + } + dim { + size: 1 + } + } + double_val: 0.9983645075733464 + } + } + } +} +node { + name: "DS_layer_type_0/bias_1_0/read" + op: "Identity" + input: "DS_layer_type_0/bias_1_0" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@DS_layer_type_0/bias_1_0" + } + } + } +} +node { + name: "DS_layer_type_0/MatMul" + op: "MatMul" + input: "DS_layer_type_0/Reshape_1" + input: "DS_layer_type_0/matrix_1_0/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "transpose_a" + value { + b: false + } + } + attr { + key: "transpose_b" + value { + b: false + } + } +} +node { + name: "DS_layer_type_0/add" + op: "Add" + input: "DS_layer_type_0/MatMul" + input: "DS_layer_type_0/bias_1_0/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "DS_layer_type_0/Tanh" + op: "Tanh" + input: "DS_layer_type_0/add" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "DS_layer_type_0/add_1" + op: "Add" + input: "DS_layer_type_0/Reshape_1" + input: "DS_layer_type_0/Tanh" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "DS_layer_type_0/Reshape_2/shape" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 3 + } + } + tensor_content: "\377\377\377\377\001\000\000\000\001\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_0/Reshape_2" + op: "Reshape" + input: "DS_layer_type_0/add_1" + input: "DS_layer_type_0/Reshape_2/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "DS_layer_type_0/Slice_2/begin" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\000\000\000\000\004\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_0/Slice_2/size" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\377\377\377\377\004\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_0/Slice_2" + op: "Slice" + input: "Reshape_4" + input: "DS_layer_type_0/Slice_2/begin" + input: "DS_layer_type_0/Slice_2/size" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "DS_layer_type_0/Reshape_3/shape" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\377\377\377\377\004\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_0/Reshape_3" + op: "Reshape" + input: "DS_layer_type_0/Slice_2" + input: "DS_layer_type_0/Reshape_3/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "DS_layer_type_0/Slice_3/begin" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\000\000\000\000\000\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_0/Slice_3/size" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\377\377\377\377\001\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_0/Slice_3" + op: "Slice" + input: "DS_layer_type_0/Reshape_3" + input: "DS_layer_type_0/Slice_3/begin" + input: "DS_layer_type_0/Slice_3/size" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "DS_layer_type_0/Reshape_4/shape" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\377\377\377\377\001\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_0/Reshape_4" + op: "Reshape" + input: "DS_layer_type_0/Slice_3" + input: "DS_layer_type_0/Reshape_4/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "DS_layer_type_0/matrix_1_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_DOUBLE + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_DOUBLE + tensor_shape { + dim { + size: 1 + } + dim { + size: 1 + } + } + double_val: 0.18976932648679823 + } + } + } +} +node { + name: "DS_layer_type_0/matrix_1_1/read" + op: "Identity" + input: "DS_layer_type_0/matrix_1_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@DS_layer_type_0/matrix_1_1" + } + } + } +} +node { + name: "DS_layer_type_0/bias_1_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_DOUBLE + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_DOUBLE + tensor_shape { + dim { + size: 1 + } + dim { + size: 1 + } + } + double_val: -0.387826139590999 + } + } + } +} +node { + name: "DS_layer_type_0/bias_1_1/read" + op: "Identity" + input: "DS_layer_type_0/bias_1_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@DS_layer_type_0/bias_1_1" + } + } + } +} +node { + name: "DS_layer_type_0/MatMul_1" + op: "MatMul" + input: "DS_layer_type_0/Reshape_4" + input: "DS_layer_type_0/matrix_1_1/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "transpose_a" + value { + b: false + } + } + attr { + key: "transpose_b" + value { + b: false + } + } +} +node { + name: "DS_layer_type_0/add_2" + op: "Add" + input: "DS_layer_type_0/MatMul_1" + input: "DS_layer_type_0/bias_1_1/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "DS_layer_type_0/Tanh_1" + op: "Tanh" + input: "DS_layer_type_0/add_2" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "DS_layer_type_0/add_3" + op: "Add" + input: "DS_layer_type_0/Reshape_4" + input: "DS_layer_type_0/Tanh_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "DS_layer_type_0/Reshape_5/shape" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 3 + } + } + tensor_content: "\377\377\377\377\001\000\000\000\001\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_0/Reshape_5" + op: "Reshape" + input: "DS_layer_type_0/add_3" + input: "DS_layer_type_0/Reshape_5/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "DS_layer_type_0/concat/axis" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "DS_layer_type_0/concat" + op: "ConcatV2" + input: "DS_layer_type_0/Reshape_2" + input: "DS_layer_type_0/Reshape_5" + input: "DS_layer_type_0/concat/axis" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } +} +node { + name: "DS_layer_type_0/Reshape_6/shape" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 3 + } + } + tensor_content: "\377\377\377\377\002\000\000\000\004\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_0/Reshape_6" + op: "Reshape" + input: "Reshape_4" + input: "DS_layer_type_0/Reshape_6/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "DS_layer_type_0/MatMul_2" + op: "BatchMatMulV2" + input: "DS_layer_type_0/Reshape_6" + input: "DS_layer_type_0/concat" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "adj_x" + value { + b: true + } + } + attr { + key: "adj_y" + value { + b: false + } + } +} +node { + name: "DS_layer_type_0/mul/y" + op: "Const" + attr { + key: "dtype" + value { + type: DT_DOUBLE + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_DOUBLE + tensor_shape { + } + double_val: 0.5 + } + } + } +} +node { + name: "DS_layer_type_0/mul" + op: "Mul" + input: "DS_layer_type_0/MatMul_2" + input: "DS_layer_type_0/mul/y" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "DS_layer_type_0/Slice_4/begin" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 3 + } + } + tensor_content: "\000\000\000\000\000\000\000\000\000\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_0/Slice_4/size" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 3 + } + } + tensor_content: "\377\377\377\377\377\377\377\377\001\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_0/Slice_4" + op: "Slice" + input: "DS_layer_type_0/mul" + input: "DS_layer_type_0/Slice_4/begin" + input: "DS_layer_type_0/Slice_4/size" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "DS_layer_type_0/MatMul_3" + op: "BatchMatMulV2" + input: "DS_layer_type_0/mul" + input: "DS_layer_type_0/Slice_4" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "adj_x" + value { + b: true + } + } + attr { + key: "adj_y" + value { + b: false + } + } +} +node { + name: "DS_layer_type_0/Reshape_7/shape" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\377\377\377\377\001\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_0/Reshape_7" + op: "Reshape" + input: "DS_layer_type_0/MatMul_3" + input: "DS_layer_type_0/Reshape_7/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "layer_0_type_0/matrix" + op: "Const" + attr { + key: "dtype" + value { + type: DT_DOUBLE + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_DOUBLE + tensor_shape { + dim { + size: 1 + } + dim { + size: 1 + } + } + double_val: 0.14726844145773643 + } + } + } +} +node { + name: "layer_0_type_0/matrix/read" + op: "Identity" + input: "layer_0_type_0/matrix" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@layer_0_type_0/matrix" + } + } + } +} +node { + name: "layer_0_type_0/bias" + op: "Const" + attr { + key: "dtype" + value { + type: DT_DOUBLE + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_DOUBLE + tensor_shape { + dim { + size: 1 + } + } + double_val: -0.36511124387384836 + } + } + } +} +node { + name: "layer_0_type_0/bias/read" + op: "Identity" + input: "layer_0_type_0/bias" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@layer_0_type_0/bias" + } + } + } +} +node { + name: "layer_0_type_0/MatMul" + op: "MatMul" + input: "DS_layer_type_0/Reshape_7" + input: "layer_0_type_0/matrix/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "transpose_a" + value { + b: false + } + } + attr { + key: "transpose_b" + value { + b: false + } + } +} +node { + name: "layer_0_type_0/add" + op: "Add" + input: "layer_0_type_0/MatMul" + input: "layer_0_type_0/bias/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "Tanh" + op: "Tanh" + input: "layer_0_type_0/add" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "final_layer_type_0/matrix" + op: "Const" + attr { + key: "dtype" + value { + type: DT_DOUBLE + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_DOUBLE + tensor_shape { + dim { + size: 1 + } + dim { + size: 1 + } + } + double_val: 0.3939480642205348 + } + } + } +} +node { + name: "final_layer_type_0/matrix/read" + op: "Identity" + input: "final_layer_type_0/matrix" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@final_layer_type_0/matrix" + } + } + } +} +node { + name: "final_layer_type_0/bias" + op: "Const" + attr { + key: "dtype" + value { + type: DT_DOUBLE + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_DOUBLE + tensor_shape { + dim { + size: 1 + } + } + double_val: -93.5768035239284 + } + } + } +} +node { + name: "final_layer_type_0/bias/read" + op: "Identity" + input: "final_layer_type_0/bias" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@final_layer_type_0/bias" + } + } + } +} +node { + name: "final_layer_type_0/MatMul" + op: "MatMul" + input: "Tanh" + input: "final_layer_type_0/matrix/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "transpose_a" + value { + b: false + } + } + attr { + key: "transpose_b" + value { + b: false + } + } +} +node { + name: "final_layer_type_0/add" + op: "Add" + input: "final_layer_type_0/MatMul" + input: "final_layer_type_0/bias/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "strided_slice_5/stack" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 2 + } + } + } +} +node { + name: "strided_slice_5/stack_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 3 + } + } + } +} +node { + name: "strided_slice_5/stack_2" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "strided_slice_5" + op: "StridedSlice" + input: "t_natoms" + input: "strided_slice_5/stack" + input: "strided_slice_5/stack_1" + input: "strided_slice_5/stack_2" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "begin_mask" + value { + i: 0 + } + } + attr { + key: "ellipsis_mask" + value { + i: 0 + } + } + attr { + key: "end_mask" + value { + i: 0 + } + } + attr { + key: "new_axis_mask" + value { + i: 0 + } + } + attr { + key: "shrink_axis_mask" + value { + i: 1 + } + } +} +node { + name: "Reshape_5/shape/0" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "Reshape_5/shape" + op: "Pack" + input: "Reshape_5/shape/0" + input: "strided_slice_5" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "axis" + value { + i: 0 + } + } +} +node { + name: "Reshape_5" + op: "Reshape" + input: "final_layer_type_0/add" + input: "Reshape_5/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "mul_3/y" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 8 + } + } + } +} +node { + name: "mul_3" + op: "Mul" + input: "add" + input: "mul_3/y" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "strided_slice_6/stack" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 3 + } + } + } +} +node { + name: "strided_slice_6/stack_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 4 + } + } + } +} +node { + name: "strided_slice_6/stack_2" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "strided_slice_6" + op: "StridedSlice" + input: "t_natoms" + input: "strided_slice_6/stack" + input: "strided_slice_6/stack_1" + input: "strided_slice_6/stack_2" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "begin_mask" + value { + i: 0 + } + } + attr { + key: "ellipsis_mask" + value { + i: 0 + } + } + attr { + key: "end_mask" + value { + i: 0 + } + } + attr { + key: "new_axis_mask" + value { + i: 0 + } + } + attr { + key: "shrink_axis_mask" + value { + i: 1 + } + } +} +node { + name: "mul_4/y" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 8 + } + } + } +} +node { + name: "mul_4" + op: "Mul" + input: "strided_slice_6" + input: "mul_4/y" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "Slice_1/begin/0" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 0 + } + } + } +} +node { + name: "Slice_1/begin" + op: "Pack" + input: "Slice_1/begin/0" + input: "mul_3" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "axis" + value { + i: 0 + } + } +} +node { + name: "Slice_1/size/0" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: -1 + } + } + } +} +node { + name: "Slice_1/size" + op: "Pack" + input: "Slice_1/size/0" + input: "mul_4" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "axis" + value { + i: 0 + } + } +} +node { + name: "Slice_1" + op: "Slice" + input: "Reshape_3" + input: "Slice_1/begin" + input: "Slice_1/size" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "Reshape_6/shape" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\377\377\377\377\010\000\000\000" + } + } + } +} +node { + name: "Reshape_6" + op: "Reshape" + input: "Slice_1" + input: "Reshape_6/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "DS_layer_type_1/Slice/begin" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\000\000\000\000\000\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_1/Slice/size" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\377\377\377\377\004\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_1/Slice" + op: "Slice" + input: "Reshape_6" + input: "DS_layer_type_1/Slice/begin" + input: "DS_layer_type_1/Slice/size" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "DS_layer_type_1/Reshape/shape" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\377\377\377\377\004\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_1/Reshape" + op: "Reshape" + input: "DS_layer_type_1/Slice" + input: "DS_layer_type_1/Reshape/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "DS_layer_type_1/Slice_1/begin" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\000\000\000\000\000\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_1/Slice_1/size" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\377\377\377\377\001\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_1/Slice_1" + op: "Slice" + input: "DS_layer_type_1/Reshape" + input: "DS_layer_type_1/Slice_1/begin" + input: "DS_layer_type_1/Slice_1/size" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "DS_layer_type_1/Reshape_1/shape" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\377\377\377\377\001\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_1/Reshape_1" + op: "Reshape" + input: "DS_layer_type_1/Slice_1" + input: "DS_layer_type_1/Reshape_1/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "DS_layer_type_1/matrix_1_0" + op: "Const" + attr { + key: "dtype" + value { + type: DT_DOUBLE + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_DOUBLE + tensor_shape { + dim { + size: 1 + } + dim { + size: 1 + } + } + double_val: -0.17935075392063676 + } + } + } +} +node { + name: "DS_layer_type_1/matrix_1_0/read" + op: "Identity" + input: "DS_layer_type_1/matrix_1_0" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@DS_layer_type_1/matrix_1_0" + } + } + } +} +node { + name: "DS_layer_type_1/bias_1_0" + op: "Const" + attr { + key: "dtype" + value { + type: DT_DOUBLE + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_DOUBLE + tensor_shape { + dim { + size: 1 + } + dim { + size: 1 + } + } + double_val: -0.5601597468396246 + } + } + } +} +node { + name: "DS_layer_type_1/bias_1_0/read" + op: "Identity" + input: "DS_layer_type_1/bias_1_0" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@DS_layer_type_1/bias_1_0" + } + } + } +} +node { + name: "DS_layer_type_1/MatMul" + op: "MatMul" + input: "DS_layer_type_1/Reshape_1" + input: "DS_layer_type_1/matrix_1_0/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "transpose_a" + value { + b: false + } + } + attr { + key: "transpose_b" + value { + b: false + } + } +} +node { + name: "DS_layer_type_1/add" + op: "Add" + input: "DS_layer_type_1/MatMul" + input: "DS_layer_type_1/bias_1_0/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "DS_layer_type_1/Tanh" + op: "Tanh" + input: "DS_layer_type_1/add" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "DS_layer_type_1/add_1" + op: "Add" + input: "DS_layer_type_1/Reshape_1" + input: "DS_layer_type_1/Tanh" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "DS_layer_type_1/Reshape_2/shape" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 3 + } + } + tensor_content: "\377\377\377\377\001\000\000\000\001\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_1/Reshape_2" + op: "Reshape" + input: "DS_layer_type_1/add_1" + input: "DS_layer_type_1/Reshape_2/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "DS_layer_type_1/Slice_2/begin" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\000\000\000\000\004\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_1/Slice_2/size" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\377\377\377\377\004\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_1/Slice_2" + op: "Slice" + input: "Reshape_6" + input: "DS_layer_type_1/Slice_2/begin" + input: "DS_layer_type_1/Slice_2/size" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "DS_layer_type_1/Reshape_3/shape" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\377\377\377\377\004\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_1/Reshape_3" + op: "Reshape" + input: "DS_layer_type_1/Slice_2" + input: "DS_layer_type_1/Reshape_3/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "DS_layer_type_1/Slice_3/begin" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\000\000\000\000\000\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_1/Slice_3/size" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\377\377\377\377\001\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_1/Slice_3" + op: "Slice" + input: "DS_layer_type_1/Reshape_3" + input: "DS_layer_type_1/Slice_3/begin" + input: "DS_layer_type_1/Slice_3/size" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "DS_layer_type_1/Reshape_4/shape" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\377\377\377\377\001\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_1/Reshape_4" + op: "Reshape" + input: "DS_layer_type_1/Slice_3" + input: "DS_layer_type_1/Reshape_4/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "DS_layer_type_1/matrix_1_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_DOUBLE + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_DOUBLE + tensor_shape { + dim { + size: 1 + } + dim { + size: 1 + } + } + double_val: 0.3205658142933548 + } + } + } +} +node { + name: "DS_layer_type_1/matrix_1_1/read" + op: "Identity" + input: "DS_layer_type_1/matrix_1_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@DS_layer_type_1/matrix_1_1" + } + } + } +} +node { + name: "DS_layer_type_1/bias_1_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_DOUBLE + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_DOUBLE + tensor_shape { + dim { + size: 1 + } + dim { + size: 1 + } + } + double_val: 0.12077221835391395 + } + } + } +} +node { + name: "DS_layer_type_1/bias_1_1/read" + op: "Identity" + input: "DS_layer_type_1/bias_1_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@DS_layer_type_1/bias_1_1" + } + } + } +} +node { + name: "DS_layer_type_1/MatMul_1" + op: "MatMul" + input: "DS_layer_type_1/Reshape_4" + input: "DS_layer_type_1/matrix_1_1/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "transpose_a" + value { + b: false + } + } + attr { + key: "transpose_b" + value { + b: false + } + } +} +node { + name: "DS_layer_type_1/add_2" + op: "Add" + input: "DS_layer_type_1/MatMul_1" + input: "DS_layer_type_1/bias_1_1/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "DS_layer_type_1/Tanh_1" + op: "Tanh" + input: "DS_layer_type_1/add_2" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "DS_layer_type_1/add_3" + op: "Add" + input: "DS_layer_type_1/Reshape_4" + input: "DS_layer_type_1/Tanh_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "DS_layer_type_1/Reshape_5/shape" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 3 + } + } + tensor_content: "\377\377\377\377\001\000\000\000\001\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_1/Reshape_5" + op: "Reshape" + input: "DS_layer_type_1/add_3" + input: "DS_layer_type_1/Reshape_5/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "DS_layer_type_1/concat/axis" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "DS_layer_type_1/concat" + op: "ConcatV2" + input: "DS_layer_type_1/Reshape_2" + input: "DS_layer_type_1/Reshape_5" + input: "DS_layer_type_1/concat/axis" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } +} +node { + name: "DS_layer_type_1/Reshape_6/shape" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 3 + } + } + tensor_content: "\377\377\377\377\002\000\000\000\004\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_1/Reshape_6" + op: "Reshape" + input: "Reshape_6" + input: "DS_layer_type_1/Reshape_6/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "DS_layer_type_1/MatMul_2" + op: "BatchMatMulV2" + input: "DS_layer_type_1/Reshape_6" + input: "DS_layer_type_1/concat" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "adj_x" + value { + b: true + } + } + attr { + key: "adj_y" + value { + b: false + } + } +} +node { + name: "DS_layer_type_1/mul/y" + op: "Const" + attr { + key: "dtype" + value { + type: DT_DOUBLE + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_DOUBLE + tensor_shape { + } + double_val: 0.5 + } + } + } +} +node { + name: "DS_layer_type_1/mul" + op: "Mul" + input: "DS_layer_type_1/MatMul_2" + input: "DS_layer_type_1/mul/y" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "DS_layer_type_1/Slice_4/begin" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 3 + } + } + tensor_content: "\000\000\000\000\000\000\000\000\000\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_1/Slice_4/size" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 3 + } + } + tensor_content: "\377\377\377\377\377\377\377\377\001\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_1/Slice_4" + op: "Slice" + input: "DS_layer_type_1/mul" + input: "DS_layer_type_1/Slice_4/begin" + input: "DS_layer_type_1/Slice_4/size" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "DS_layer_type_1/MatMul_3" + op: "BatchMatMulV2" + input: "DS_layer_type_1/mul" + input: "DS_layer_type_1/Slice_4" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "adj_x" + value { + b: true + } + } + attr { + key: "adj_y" + value { + b: false + } + } +} +node { + name: "DS_layer_type_1/Reshape_7/shape" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\377\377\377\377\001\000\000\000" + } + } + } +} +node { + name: "DS_layer_type_1/Reshape_7" + op: "Reshape" + input: "DS_layer_type_1/MatMul_3" + input: "DS_layer_type_1/Reshape_7/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "layer_0_type_1/matrix" + op: "Const" + attr { + key: "dtype" + value { + type: DT_DOUBLE + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_DOUBLE + tensor_shape { + dim { + size: 1 + } + dim { + size: 1 + } + } + double_val: 0.1934560948696963 + } + } + } +} +node { + name: "layer_0_type_1/matrix/read" + op: "Identity" + input: "layer_0_type_1/matrix" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@layer_0_type_1/matrix" + } + } + } +} +node { + name: "layer_0_type_1/bias" + op: "Const" + attr { + key: "dtype" + value { + type: DT_DOUBLE + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_DOUBLE + tensor_shape { + dim { + size: 1 + } + } + double_val: -0.17107361582243172 + } + } + } +} +node { + name: "layer_0_type_1/bias/read" + op: "Identity" + input: "layer_0_type_1/bias" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@layer_0_type_1/bias" + } + } + } +} +node { + name: "layer_0_type_1/MatMul" + op: "MatMul" + input: "DS_layer_type_1/Reshape_7" + input: "layer_0_type_1/matrix/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "transpose_a" + value { + b: false + } + } + attr { + key: "transpose_b" + value { + b: false + } + } +} +node { + name: "layer_0_type_1/add" + op: "Add" + input: "layer_0_type_1/MatMul" + input: "layer_0_type_1/bias/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "Tanh_1" + op: "Tanh" + input: "layer_0_type_1/add" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "final_layer_type_1/matrix" + op: "Const" + attr { + key: "dtype" + value { + type: DT_DOUBLE + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_DOUBLE + tensor_shape { + dim { + size: 1 + } + dim { + size: 1 + } + } + double_val: 0.3119468727172958 + } + } + } +} +node { + name: "final_layer_type_1/matrix/read" + op: "Identity" + input: "final_layer_type_1/matrix" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@final_layer_type_1/matrix" + } + } + } +} +node { + name: "final_layer_type_1/bias" + op: "Const" + attr { + key: "dtype" + value { + type: DT_DOUBLE + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_DOUBLE + tensor_shape { + dim { + size: 1 + } + } + double_val: -187.15053100832083 + } + } + } +} +node { + name: "final_layer_type_1/bias/read" + op: "Identity" + input: "final_layer_type_1/bias" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@final_layer_type_1/bias" + } + } + } +} +node { + name: "final_layer_type_1/MatMul" + op: "MatMul" + input: "Tanh_1" + input: "final_layer_type_1/matrix/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "transpose_a" + value { + b: false + } + } + attr { + key: "transpose_b" + value { + b: false + } + } +} +node { + name: "final_layer_type_1/add" + op: "Add" + input: "final_layer_type_1/MatMul" + input: "final_layer_type_1/bias/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "strided_slice_8/stack" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 3 + } + } + } +} +node { + name: "strided_slice_8/stack_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 4 + } + } + } +} +node { + name: "strided_slice_8/stack_2" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "strided_slice_8" + op: "StridedSlice" + input: "t_natoms" + input: "strided_slice_8/stack" + input: "strided_slice_8/stack_1" + input: "strided_slice_8/stack_2" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "begin_mask" + value { + i: 0 + } + } + attr { + key: "ellipsis_mask" + value { + i: 0 + } + } + attr { + key: "end_mask" + value { + i: 0 + } + } + attr { + key: "new_axis_mask" + value { + i: 0 + } + } + attr { + key: "shrink_axis_mask" + value { + i: 1 + } + } +} +node { + name: "Reshape_7/shape/0" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "Reshape_7/shape" + op: "Pack" + input: "Reshape_7/shape/0" + input: "strided_slice_8" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "axis" + value { + i: 0 + } + } +} +node { + name: "Reshape_7" + op: "Reshape" + input: "final_layer_type_1/add" + input: "Reshape_7/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "concat/axis" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "concat" + op: "ConcatV2" + input: "Reshape_5" + input: "Reshape_7" + input: "concat/axis" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } +} +node { + name: "Reshape_8/shape" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: -1 + } + } + } +} +node { + name: "Reshape_8" + op: "Reshape" + input: "concat" + input: "Reshape_8/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "strided_slice_9/stack" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 0 + } + } + } +} +node { + name: "strided_slice_9/stack_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "strided_slice_9/stack_2" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "strided_slice_9" + op: "StridedSlice" + input: "t_natoms" + input: "strided_slice_9/stack" + input: "strided_slice_9/stack_1" + input: "strided_slice_9/stack_2" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "begin_mask" + value { + i: 0 + } + } + attr { + key: "ellipsis_mask" + value { + i: 0 + } + } + attr { + key: "end_mask" + value { + i: 0 + } + } + attr { + key: "new_axis_mask" + value { + i: 0 + } + } + attr { + key: "shrink_axis_mask" + value { + i: 1 + } + } +} +node { + name: "atom_energy_test/shape/0" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: -1 + } + } + } +} +node { + name: "atom_energy_test/shape" + op: "Pack" + input: "atom_energy_test/shape/0" + input: "strided_slice_9" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "axis" + value { + i: 0 + } + } +} +node { + name: "atom_energy_test" + op: "Reshape" + input: "Reshape_8" + input: "atom_energy_test/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "energy_test/reduction_indices" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "energy_test" + op: "Sum" + input: "atom_energy_test" + input: "energy_test/reduction_indices" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/Shape" + op: "Shape" + input: "Reshape_8" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/grad_ys_0" + op: "Const" + attr { + key: "dtype" + value { + type: DT_DOUBLE + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_DOUBLE + tensor_shape { + } + double_val: 1.0 + } + } + } +} +node { + name: "gradients/Fill" + op: "Fill" + input: "gradients/Shape" + input: "gradients/grad_ys_0" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "index_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/Reshape_8_grad/Shape" + op: "Shape" + input: "concat" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/Reshape_8_grad/Reshape" + op: "Reshape" + input: "gradients/Fill" + input: "gradients/Reshape_8_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/concat_grad/Rank" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 2 + } + } + } +} +node { + name: "gradients/concat_grad/mod" + op: "FloorMod" + input: "concat/axis" + input: "gradients/concat_grad/Rank" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/concat_grad/ShapeN" + op: "ShapeN" + input: "Reshape_5" + input: "Reshape_7" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/concat_grad/ConcatOffset" + op: "ConcatOffset" + input: "gradients/concat_grad/mod" + input: "gradients/concat_grad/ShapeN" + input: "gradients/concat_grad/ShapeN:1" + attr { + key: "N" + value { + i: 2 + } + } +} +node { + name: "gradients/concat_grad/Slice" + op: "Slice" + input: "gradients/Reshape_8_grad/Reshape" + input: "gradients/concat_grad/ConcatOffset" + input: "gradients/concat_grad/ShapeN" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "gradients/concat_grad/Slice_1" + op: "Slice" + input: "gradients/Reshape_8_grad/Reshape" + input: "gradients/concat_grad/ConcatOffset:1" + input: "gradients/concat_grad/ShapeN:1" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "gradients/Reshape_5_grad/Shape" + op: "Shape" + input: "final_layer_type_0/add" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/Reshape_5_grad/Reshape" + op: "Reshape" + input: "gradients/concat_grad/Slice" + input: "gradients/Reshape_5_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/Reshape_7_grad/Shape" + op: "Shape" + input: "final_layer_type_1/add" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/Reshape_7_grad/Reshape" + op: "Reshape" + input: "gradients/concat_grad/Slice_1" + input: "gradients/Reshape_7_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/final_layer_type_0/add_grad/Shape" + op: "Shape" + input: "final_layer_type_0/MatMul" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/final_layer_type_0/add_grad/Shape_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "gradients/final_layer_type_0/add_grad/BroadcastGradientArgs" + op: "BroadcastGradientArgs" + input: "gradients/final_layer_type_0/add_grad/Shape" + input: "gradients/final_layer_type_0/add_grad/Shape_1" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/final_layer_type_0/add_grad/Sum" + op: "Sum" + input: "gradients/Reshape_5_grad/Reshape" + input: "gradients/final_layer_type_0/add_grad/BroadcastGradientArgs" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/final_layer_type_0/add_grad/Reshape" + op: "Reshape" + input: "gradients/final_layer_type_0/add_grad/Sum" + input: "gradients/final_layer_type_0/add_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/final_layer_type_1/add_grad/Shape" + op: "Shape" + input: "final_layer_type_1/MatMul" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/final_layer_type_1/add_grad/Shape_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "gradients/final_layer_type_1/add_grad/BroadcastGradientArgs" + op: "BroadcastGradientArgs" + input: "gradients/final_layer_type_1/add_grad/Shape" + input: "gradients/final_layer_type_1/add_grad/Shape_1" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/final_layer_type_1/add_grad/Sum" + op: "Sum" + input: "gradients/Reshape_7_grad/Reshape" + input: "gradients/final_layer_type_1/add_grad/BroadcastGradientArgs" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/final_layer_type_1/add_grad/Reshape" + op: "Reshape" + input: "gradients/final_layer_type_1/add_grad/Sum" + input: "gradients/final_layer_type_1/add_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/final_layer_type_0/MatMul_grad/MatMul" + op: "MatMul" + input: "gradients/final_layer_type_0/add_grad/Reshape" + input: "final_layer_type_0/matrix/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "transpose_a" + value { + b: false + } + } + attr { + key: "transpose_b" + value { + b: true + } + } +} +node { + name: "gradients/final_layer_type_1/MatMul_grad/MatMul" + op: "MatMul" + input: "gradients/final_layer_type_1/add_grad/Reshape" + input: "final_layer_type_1/matrix/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "transpose_a" + value { + b: false + } + } + attr { + key: "transpose_b" + value { + b: true + } + } +} +node { + name: "gradients/Tanh_grad/TanhGrad" + op: "TanhGrad" + input: "Tanh" + input: "gradients/final_layer_type_0/MatMul_grad/MatMul" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "gradients/Tanh_1_grad/TanhGrad" + op: "TanhGrad" + input: "Tanh_1" + input: "gradients/final_layer_type_1/MatMul_grad/MatMul" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "gradients/layer_0_type_0/add_grad/Shape" + op: "Shape" + input: "layer_0_type_0/MatMul" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/layer_0_type_0/add_grad/Shape_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "gradients/layer_0_type_0/add_grad/BroadcastGradientArgs" + op: "BroadcastGradientArgs" + input: "gradients/layer_0_type_0/add_grad/Shape" + input: "gradients/layer_0_type_0/add_grad/Shape_1" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/layer_0_type_0/add_grad/Sum" + op: "Sum" + input: "gradients/Tanh_grad/TanhGrad" + input: "gradients/layer_0_type_0/add_grad/BroadcastGradientArgs" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/layer_0_type_0/add_grad/Reshape" + op: "Reshape" + input: "gradients/layer_0_type_0/add_grad/Sum" + input: "gradients/layer_0_type_0/add_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/layer_0_type_1/add_grad/Shape" + op: "Shape" + input: "layer_0_type_1/MatMul" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/layer_0_type_1/add_grad/Shape_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "gradients/layer_0_type_1/add_grad/BroadcastGradientArgs" + op: "BroadcastGradientArgs" + input: "gradients/layer_0_type_1/add_grad/Shape" + input: "gradients/layer_0_type_1/add_grad/Shape_1" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/layer_0_type_1/add_grad/Sum" + op: "Sum" + input: "gradients/Tanh_1_grad/TanhGrad" + input: "gradients/layer_0_type_1/add_grad/BroadcastGradientArgs" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/layer_0_type_1/add_grad/Reshape" + op: "Reshape" + input: "gradients/layer_0_type_1/add_grad/Sum" + input: "gradients/layer_0_type_1/add_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/layer_0_type_0/MatMul_grad/MatMul" + op: "MatMul" + input: "gradients/layer_0_type_0/add_grad/Reshape" + input: "layer_0_type_0/matrix/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "transpose_a" + value { + b: false + } + } + attr { + key: "transpose_b" + value { + b: true + } + } +} +node { + name: "gradients/layer_0_type_1/MatMul_grad/MatMul" + op: "MatMul" + input: "gradients/layer_0_type_1/add_grad/Reshape" + input: "layer_0_type_1/matrix/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "transpose_a" + value { + b: false + } + } + attr { + key: "transpose_b" + value { + b: true + } + } +} +node { + name: "gradients/DS_layer_type_0/Reshape_7_grad/Shape" + op: "Shape" + input: "DS_layer_type_0/MatMul_3" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Reshape_7_grad/Reshape" + op: "Reshape" + input: "gradients/layer_0_type_0/MatMul_grad/MatMul" + input: "gradients/DS_layer_type_0/Reshape_7_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Reshape_7_grad/Shape" + op: "Shape" + input: "DS_layer_type_1/MatMul_3" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Reshape_7_grad/Reshape" + op: "Reshape" + input: "gradients/layer_0_type_1/MatMul_grad/MatMul" + input: "gradients/DS_layer_type_1/Reshape_7_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_3_grad/MatMul" + op: "BatchMatMulV2" + input: "DS_layer_type_0/Slice_4" + input: "gradients/DS_layer_type_0/Reshape_7_grad/Reshape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "adj_x" + value { + b: false + } + } + attr { + key: "adj_y" + value { + b: true + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_3_grad/MatMul_1" + op: "BatchMatMulV2" + input: "DS_layer_type_0/mul" + input: "gradients/DS_layer_type_0/Reshape_7_grad/Reshape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "adj_x" + value { + b: false + } + } + attr { + key: "adj_y" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_3_grad/Shape" + op: "Shape" + input: "DS_layer_type_0/mul" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_3_grad/Shape_1" + op: "Shape" + input: "DS_layer_type_0/Slice_4" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_3_grad/strided_slice/stack" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 0 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_3_grad/strided_slice/stack_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: -2 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_3_grad/strided_slice/stack_2" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_3_grad/strided_slice" + op: "StridedSlice" + input: "gradients/DS_layer_type_0/MatMul_3_grad/Shape" + input: "gradients/DS_layer_type_0/MatMul_3_grad/strided_slice/stack" + input: "gradients/DS_layer_type_0/MatMul_3_grad/strided_slice/stack_1" + input: "gradients/DS_layer_type_0/MatMul_3_grad/strided_slice/stack_2" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "begin_mask" + value { + i: 1 + } + } + attr { + key: "ellipsis_mask" + value { + i: 0 + } + } + attr { + key: "end_mask" + value { + i: 0 + } + } + attr { + key: "new_axis_mask" + value { + i: 0 + } + } + attr { + key: "shrink_axis_mask" + value { + i: 0 + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_3_grad/strided_slice_1/stack" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 0 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_3_grad/strided_slice_1/stack_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: -2 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_3_grad/strided_slice_1/stack_2" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_3_grad/strided_slice_1" + op: "StridedSlice" + input: "gradients/DS_layer_type_0/MatMul_3_grad/Shape_1" + input: "gradients/DS_layer_type_0/MatMul_3_grad/strided_slice_1/stack" + input: "gradients/DS_layer_type_0/MatMul_3_grad/strided_slice_1/stack_1" + input: "gradients/DS_layer_type_0/MatMul_3_grad/strided_slice_1/stack_2" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "begin_mask" + value { + i: 1 + } + } + attr { + key: "ellipsis_mask" + value { + i: 0 + } + } + attr { + key: "end_mask" + value { + i: 0 + } + } + attr { + key: "new_axis_mask" + value { + i: 0 + } + } + attr { + key: "shrink_axis_mask" + value { + i: 0 + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_3_grad/BroadcastGradientArgs" + op: "BroadcastGradientArgs" + input: "gradients/DS_layer_type_0/MatMul_3_grad/strided_slice" + input: "gradients/DS_layer_type_0/MatMul_3_grad/strided_slice_1" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_3_grad/Sum" + op: "Sum" + input: "gradients/DS_layer_type_0/MatMul_3_grad/MatMul" + input: "gradients/DS_layer_type_0/MatMul_3_grad/BroadcastGradientArgs" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_3_grad/Reshape" + op: "Reshape" + input: "gradients/DS_layer_type_0/MatMul_3_grad/Sum" + input: "gradients/DS_layer_type_0/MatMul_3_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_3_grad/Sum_1" + op: "Sum" + input: "gradients/DS_layer_type_0/MatMul_3_grad/MatMul_1" + input: "gradients/DS_layer_type_0/MatMul_3_grad/BroadcastGradientArgs:1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_3_grad/Reshape_1" + op: "Reshape" + input: "gradients/DS_layer_type_0/MatMul_3_grad/Sum_1" + input: "gradients/DS_layer_type_0/MatMul_3_grad/Shape_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_3_grad/MatMul" + op: "BatchMatMulV2" + input: "DS_layer_type_1/Slice_4" + input: "gradients/DS_layer_type_1/Reshape_7_grad/Reshape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "adj_x" + value { + b: false + } + } + attr { + key: "adj_y" + value { + b: true + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_3_grad/MatMul_1" + op: "BatchMatMulV2" + input: "DS_layer_type_1/mul" + input: "gradients/DS_layer_type_1/Reshape_7_grad/Reshape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "adj_x" + value { + b: false + } + } + attr { + key: "adj_y" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_3_grad/Shape" + op: "Shape" + input: "DS_layer_type_1/mul" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_3_grad/Shape_1" + op: "Shape" + input: "DS_layer_type_1/Slice_4" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_3_grad/strided_slice/stack" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 0 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_3_grad/strided_slice/stack_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: -2 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_3_grad/strided_slice/stack_2" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_3_grad/strided_slice" + op: "StridedSlice" + input: "gradients/DS_layer_type_1/MatMul_3_grad/Shape" + input: "gradients/DS_layer_type_1/MatMul_3_grad/strided_slice/stack" + input: "gradients/DS_layer_type_1/MatMul_3_grad/strided_slice/stack_1" + input: "gradients/DS_layer_type_1/MatMul_3_grad/strided_slice/stack_2" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "begin_mask" + value { + i: 1 + } + } + attr { + key: "ellipsis_mask" + value { + i: 0 + } + } + attr { + key: "end_mask" + value { + i: 0 + } + } + attr { + key: "new_axis_mask" + value { + i: 0 + } + } + attr { + key: "shrink_axis_mask" + value { + i: 0 + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_3_grad/strided_slice_1/stack" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 0 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_3_grad/strided_slice_1/stack_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: -2 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_3_grad/strided_slice_1/stack_2" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_3_grad/strided_slice_1" + op: "StridedSlice" + input: "gradients/DS_layer_type_1/MatMul_3_grad/Shape_1" + input: "gradients/DS_layer_type_1/MatMul_3_grad/strided_slice_1/stack" + input: "gradients/DS_layer_type_1/MatMul_3_grad/strided_slice_1/stack_1" + input: "gradients/DS_layer_type_1/MatMul_3_grad/strided_slice_1/stack_2" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "begin_mask" + value { + i: 1 + } + } + attr { + key: "ellipsis_mask" + value { + i: 0 + } + } + attr { + key: "end_mask" + value { + i: 0 + } + } + attr { + key: "new_axis_mask" + value { + i: 0 + } + } + attr { + key: "shrink_axis_mask" + value { + i: 0 + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_3_grad/BroadcastGradientArgs" + op: "BroadcastGradientArgs" + input: "gradients/DS_layer_type_1/MatMul_3_grad/strided_slice" + input: "gradients/DS_layer_type_1/MatMul_3_grad/strided_slice_1" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_3_grad/Sum" + op: "Sum" + input: "gradients/DS_layer_type_1/MatMul_3_grad/MatMul" + input: "gradients/DS_layer_type_1/MatMul_3_grad/BroadcastGradientArgs" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_3_grad/Reshape" + op: "Reshape" + input: "gradients/DS_layer_type_1/MatMul_3_grad/Sum" + input: "gradients/DS_layer_type_1/MatMul_3_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_3_grad/Sum_1" + op: "Sum" + input: "gradients/DS_layer_type_1/MatMul_3_grad/MatMul_1" + input: "gradients/DS_layer_type_1/MatMul_3_grad/BroadcastGradientArgs:1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_3_grad/Reshape_1" + op: "Reshape" + input: "gradients/DS_layer_type_1/MatMul_3_grad/Sum_1" + input: "gradients/DS_layer_type_1/MatMul_3_grad/Shape_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_4_grad/Rank" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 3 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_4_grad/Shape" + op: "Shape" + input: "DS_layer_type_0/Slice_4" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_4_grad/stack/1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_4_grad/stack" + op: "Pack" + input: "gradients/DS_layer_type_0/Slice_4_grad/Rank" + input: "gradients/DS_layer_type_0/Slice_4_grad/stack/1" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "axis" + value { + i: 0 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_4_grad/Reshape" + op: "Reshape" + input: "DS_layer_type_0/Slice_4/begin" + input: "gradients/DS_layer_type_0/Slice_4_grad/stack" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_4_grad/Shape_1" + op: "Shape" + input: "DS_layer_type_0/mul" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_4_grad/sub" + op: "Sub" + input: "gradients/DS_layer_type_0/Slice_4_grad/Shape_1" + input: "gradients/DS_layer_type_0/Slice_4_grad/Shape" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_4_grad/sub_1" + op: "Sub" + input: "gradients/DS_layer_type_0/Slice_4_grad/sub" + input: "DS_layer_type_0/Slice_4/begin" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_4_grad/Reshape_1" + op: "Reshape" + input: "gradients/DS_layer_type_0/Slice_4_grad/sub_1" + input: "gradients/DS_layer_type_0/Slice_4_grad/stack" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_4_grad/concat/axis" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_4_grad/concat" + op: "ConcatV2" + input: "gradients/DS_layer_type_0/Slice_4_grad/Reshape" + input: "gradients/DS_layer_type_0/Slice_4_grad/Reshape_1" + input: "gradients/DS_layer_type_0/Slice_4_grad/concat/axis" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_4_grad/Pad" + op: "Pad" + input: "gradients/DS_layer_type_0/MatMul_3_grad/Reshape_1" + input: "gradients/DS_layer_type_0/Slice_4_grad/concat" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tpaddings" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_4_grad/Rank" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 3 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_4_grad/Shape" + op: "Shape" + input: "DS_layer_type_1/Slice_4" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_4_grad/stack/1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_4_grad/stack" + op: "Pack" + input: "gradients/DS_layer_type_1/Slice_4_grad/Rank" + input: "gradients/DS_layer_type_1/Slice_4_grad/stack/1" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "axis" + value { + i: 0 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_4_grad/Reshape" + op: "Reshape" + input: "DS_layer_type_1/Slice_4/begin" + input: "gradients/DS_layer_type_1/Slice_4_grad/stack" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_4_grad/Shape_1" + op: "Shape" + input: "DS_layer_type_1/mul" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_4_grad/sub" + op: "Sub" + input: "gradients/DS_layer_type_1/Slice_4_grad/Shape_1" + input: "gradients/DS_layer_type_1/Slice_4_grad/Shape" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_4_grad/sub_1" + op: "Sub" + input: "gradients/DS_layer_type_1/Slice_4_grad/sub" + input: "DS_layer_type_1/Slice_4/begin" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_4_grad/Reshape_1" + op: "Reshape" + input: "gradients/DS_layer_type_1/Slice_4_grad/sub_1" + input: "gradients/DS_layer_type_1/Slice_4_grad/stack" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_4_grad/concat/axis" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_4_grad/concat" + op: "ConcatV2" + input: "gradients/DS_layer_type_1/Slice_4_grad/Reshape" + input: "gradients/DS_layer_type_1/Slice_4_grad/Reshape_1" + input: "gradients/DS_layer_type_1/Slice_4_grad/concat/axis" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_4_grad/Pad" + op: "Pad" + input: "gradients/DS_layer_type_1/MatMul_3_grad/Reshape_1" + input: "gradients/DS_layer_type_1/Slice_4_grad/concat" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tpaddings" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/AddN" + op: "AddN" + input: "gradients/DS_layer_type_0/MatMul_3_grad/Reshape" + input: "gradients/DS_layer_type_0/Slice_4_grad/Pad" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@gradients/DS_layer_type_0/MatMul_3_grad/Reshape" + } + } + } +} +node { + name: "gradients/DS_layer_type_0/mul_grad/Shape" + op: "Shape" + input: "DS_layer_type_0/MatMul_2" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/mul_grad/Shape_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + } + } + } + } + } +} +node { + name: "gradients/DS_layer_type_0/mul_grad/BroadcastGradientArgs" + op: "BroadcastGradientArgs" + input: "gradients/DS_layer_type_0/mul_grad/Shape" + input: "gradients/DS_layer_type_0/mul_grad/Shape_1" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/mul_grad/Mul" + op: "Mul" + input: "gradients/AddN" + input: "DS_layer_type_0/mul/y" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "gradients/DS_layer_type_0/mul_grad/Sum" + op: "Sum" + input: "gradients/DS_layer_type_0/mul_grad/Mul" + input: "gradients/DS_layer_type_0/mul_grad/BroadcastGradientArgs" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_0/mul_grad/Reshape" + op: "Reshape" + input: "gradients/DS_layer_type_0/mul_grad/Sum" + input: "gradients/DS_layer_type_0/mul_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/AddN_1" + op: "AddN" + input: "gradients/DS_layer_type_1/MatMul_3_grad/Reshape" + input: "gradients/DS_layer_type_1/Slice_4_grad/Pad" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@gradients/DS_layer_type_1/MatMul_3_grad/Reshape" + } + } + } +} +node { + name: "gradients/DS_layer_type_1/mul_grad/Shape" + op: "Shape" + input: "DS_layer_type_1/MatMul_2" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/mul_grad/Shape_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + } + } + } + } + } +} +node { + name: "gradients/DS_layer_type_1/mul_grad/BroadcastGradientArgs" + op: "BroadcastGradientArgs" + input: "gradients/DS_layer_type_1/mul_grad/Shape" + input: "gradients/DS_layer_type_1/mul_grad/Shape_1" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/mul_grad/Mul" + op: "Mul" + input: "gradients/AddN_1" + input: "DS_layer_type_1/mul/y" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "gradients/DS_layer_type_1/mul_grad/Sum" + op: "Sum" + input: "gradients/DS_layer_type_1/mul_grad/Mul" + input: "gradients/DS_layer_type_1/mul_grad/BroadcastGradientArgs" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_1/mul_grad/Reshape" + op: "Reshape" + input: "gradients/DS_layer_type_1/mul_grad/Sum" + input: "gradients/DS_layer_type_1/mul_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_2_grad/MatMul" + op: "BatchMatMulV2" + input: "DS_layer_type_0/concat" + input: "gradients/DS_layer_type_0/mul_grad/Reshape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "adj_x" + value { + b: false + } + } + attr { + key: "adj_y" + value { + b: true + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_2_grad/MatMul_1" + op: "BatchMatMulV2" + input: "DS_layer_type_0/Reshape_6" + input: "gradients/DS_layer_type_0/mul_grad/Reshape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "adj_x" + value { + b: false + } + } + attr { + key: "adj_y" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_2_grad/Shape" + op: "Shape" + input: "DS_layer_type_0/Reshape_6" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_2_grad/Shape_1" + op: "Shape" + input: "DS_layer_type_0/concat" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_2_grad/strided_slice/stack" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 0 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_2_grad/strided_slice/stack_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: -2 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_2_grad/strided_slice/stack_2" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_2_grad/strided_slice" + op: "StridedSlice" + input: "gradients/DS_layer_type_0/MatMul_2_grad/Shape" + input: "gradients/DS_layer_type_0/MatMul_2_grad/strided_slice/stack" + input: "gradients/DS_layer_type_0/MatMul_2_grad/strided_slice/stack_1" + input: "gradients/DS_layer_type_0/MatMul_2_grad/strided_slice/stack_2" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "begin_mask" + value { + i: 1 + } + } + attr { + key: "ellipsis_mask" + value { + i: 0 + } + } + attr { + key: "end_mask" + value { + i: 0 + } + } + attr { + key: "new_axis_mask" + value { + i: 0 + } + } + attr { + key: "shrink_axis_mask" + value { + i: 0 + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_2_grad/strided_slice_1/stack" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 0 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_2_grad/strided_slice_1/stack_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: -2 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_2_grad/strided_slice_1/stack_2" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_2_grad/strided_slice_1" + op: "StridedSlice" + input: "gradients/DS_layer_type_0/MatMul_2_grad/Shape_1" + input: "gradients/DS_layer_type_0/MatMul_2_grad/strided_slice_1/stack" + input: "gradients/DS_layer_type_0/MatMul_2_grad/strided_slice_1/stack_1" + input: "gradients/DS_layer_type_0/MatMul_2_grad/strided_slice_1/stack_2" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "begin_mask" + value { + i: 1 + } + } + attr { + key: "ellipsis_mask" + value { + i: 0 + } + } + attr { + key: "end_mask" + value { + i: 0 + } + } + attr { + key: "new_axis_mask" + value { + i: 0 + } + } + attr { + key: "shrink_axis_mask" + value { + i: 0 + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_2_grad/BroadcastGradientArgs" + op: "BroadcastGradientArgs" + input: "gradients/DS_layer_type_0/MatMul_2_grad/strided_slice" + input: "gradients/DS_layer_type_0/MatMul_2_grad/strided_slice_1" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_2_grad/Sum" + op: "Sum" + input: "gradients/DS_layer_type_0/MatMul_2_grad/MatMul" + input: "gradients/DS_layer_type_0/MatMul_2_grad/BroadcastGradientArgs" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_2_grad/Reshape" + op: "Reshape" + input: "gradients/DS_layer_type_0/MatMul_2_grad/Sum" + input: "gradients/DS_layer_type_0/MatMul_2_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_2_grad/Sum_1" + op: "Sum" + input: "gradients/DS_layer_type_0/MatMul_2_grad/MatMul_1" + input: "gradients/DS_layer_type_0/MatMul_2_grad/BroadcastGradientArgs:1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_2_grad/Reshape_1" + op: "Reshape" + input: "gradients/DS_layer_type_0/MatMul_2_grad/Sum_1" + input: "gradients/DS_layer_type_0/MatMul_2_grad/Shape_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_2_grad/MatMul" + op: "BatchMatMulV2" + input: "DS_layer_type_1/concat" + input: "gradients/DS_layer_type_1/mul_grad/Reshape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "adj_x" + value { + b: false + } + } + attr { + key: "adj_y" + value { + b: true + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_2_grad/MatMul_1" + op: "BatchMatMulV2" + input: "DS_layer_type_1/Reshape_6" + input: "gradients/DS_layer_type_1/mul_grad/Reshape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "adj_x" + value { + b: false + } + } + attr { + key: "adj_y" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_2_grad/Shape" + op: "Shape" + input: "DS_layer_type_1/Reshape_6" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_2_grad/Shape_1" + op: "Shape" + input: "DS_layer_type_1/concat" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_2_grad/strided_slice/stack" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 0 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_2_grad/strided_slice/stack_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: -2 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_2_grad/strided_slice/stack_2" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_2_grad/strided_slice" + op: "StridedSlice" + input: "gradients/DS_layer_type_1/MatMul_2_grad/Shape" + input: "gradients/DS_layer_type_1/MatMul_2_grad/strided_slice/stack" + input: "gradients/DS_layer_type_1/MatMul_2_grad/strided_slice/stack_1" + input: "gradients/DS_layer_type_1/MatMul_2_grad/strided_slice/stack_2" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "begin_mask" + value { + i: 1 + } + } + attr { + key: "ellipsis_mask" + value { + i: 0 + } + } + attr { + key: "end_mask" + value { + i: 0 + } + } + attr { + key: "new_axis_mask" + value { + i: 0 + } + } + attr { + key: "shrink_axis_mask" + value { + i: 0 + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_2_grad/strided_slice_1/stack" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 0 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_2_grad/strided_slice_1/stack_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: -2 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_2_grad/strided_slice_1/stack_2" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_2_grad/strided_slice_1" + op: "StridedSlice" + input: "gradients/DS_layer_type_1/MatMul_2_grad/Shape_1" + input: "gradients/DS_layer_type_1/MatMul_2_grad/strided_slice_1/stack" + input: "gradients/DS_layer_type_1/MatMul_2_grad/strided_slice_1/stack_1" + input: "gradients/DS_layer_type_1/MatMul_2_grad/strided_slice_1/stack_2" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "begin_mask" + value { + i: 1 + } + } + attr { + key: "ellipsis_mask" + value { + i: 0 + } + } + attr { + key: "end_mask" + value { + i: 0 + } + } + attr { + key: "new_axis_mask" + value { + i: 0 + } + } + attr { + key: "shrink_axis_mask" + value { + i: 0 + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_2_grad/BroadcastGradientArgs" + op: "BroadcastGradientArgs" + input: "gradients/DS_layer_type_1/MatMul_2_grad/strided_slice" + input: "gradients/DS_layer_type_1/MatMul_2_grad/strided_slice_1" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_2_grad/Sum" + op: "Sum" + input: "gradients/DS_layer_type_1/MatMul_2_grad/MatMul" + input: "gradients/DS_layer_type_1/MatMul_2_grad/BroadcastGradientArgs" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_2_grad/Reshape" + op: "Reshape" + input: "gradients/DS_layer_type_1/MatMul_2_grad/Sum" + input: "gradients/DS_layer_type_1/MatMul_2_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_2_grad/Sum_1" + op: "Sum" + input: "gradients/DS_layer_type_1/MatMul_2_grad/MatMul_1" + input: "gradients/DS_layer_type_1/MatMul_2_grad/BroadcastGradientArgs:1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_2_grad/Reshape_1" + op: "Reshape" + input: "gradients/DS_layer_type_1/MatMul_2_grad/Sum_1" + input: "gradients/DS_layer_type_1/MatMul_2_grad/Shape_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Reshape_6_grad/Shape" + op: "Shape" + input: "Reshape_4" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Reshape_6_grad/Reshape" + op: "Reshape" + input: "gradients/DS_layer_type_0/MatMul_2_grad/Reshape" + input: "gradients/DS_layer_type_0/Reshape_6_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/concat_grad/Rank" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 3 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/concat_grad/mod" + op: "FloorMod" + input: "DS_layer_type_0/concat/axis" + input: "gradients/DS_layer_type_0/concat_grad/Rank" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/concat_grad/ShapeN" + op: "ShapeN" + input: "DS_layer_type_0/Reshape_2" + input: "DS_layer_type_0/Reshape_5" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/concat_grad/ConcatOffset" + op: "ConcatOffset" + input: "gradients/DS_layer_type_0/concat_grad/mod" + input: "gradients/DS_layer_type_0/concat_grad/ShapeN" + input: "gradients/DS_layer_type_0/concat_grad/ShapeN:1" + attr { + key: "N" + value { + i: 2 + } + } +} +node { + name: "gradients/DS_layer_type_0/concat_grad/Slice" + op: "Slice" + input: "gradients/DS_layer_type_0/MatMul_2_grad/Reshape_1" + input: "gradients/DS_layer_type_0/concat_grad/ConcatOffset" + input: "gradients/DS_layer_type_0/concat_grad/ShapeN" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "gradients/DS_layer_type_0/concat_grad/Slice_1" + op: "Slice" + input: "gradients/DS_layer_type_0/MatMul_2_grad/Reshape_1" + input: "gradients/DS_layer_type_0/concat_grad/ConcatOffset:1" + input: "gradients/DS_layer_type_0/concat_grad/ShapeN:1" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "gradients/DS_layer_type_1/Reshape_6_grad/Shape" + op: "Shape" + input: "Reshape_6" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Reshape_6_grad/Reshape" + op: "Reshape" + input: "gradients/DS_layer_type_1/MatMul_2_grad/Reshape" + input: "gradients/DS_layer_type_1/Reshape_6_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/concat_grad/Rank" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 3 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/concat_grad/mod" + op: "FloorMod" + input: "DS_layer_type_1/concat/axis" + input: "gradients/DS_layer_type_1/concat_grad/Rank" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/concat_grad/ShapeN" + op: "ShapeN" + input: "DS_layer_type_1/Reshape_2" + input: "DS_layer_type_1/Reshape_5" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/concat_grad/ConcatOffset" + op: "ConcatOffset" + input: "gradients/DS_layer_type_1/concat_grad/mod" + input: "gradients/DS_layer_type_1/concat_grad/ShapeN" + input: "gradients/DS_layer_type_1/concat_grad/ShapeN:1" + attr { + key: "N" + value { + i: 2 + } + } +} +node { + name: "gradients/DS_layer_type_1/concat_grad/Slice" + op: "Slice" + input: "gradients/DS_layer_type_1/MatMul_2_grad/Reshape_1" + input: "gradients/DS_layer_type_1/concat_grad/ConcatOffset" + input: "gradients/DS_layer_type_1/concat_grad/ShapeN" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "gradients/DS_layer_type_1/concat_grad/Slice_1" + op: "Slice" + input: "gradients/DS_layer_type_1/MatMul_2_grad/Reshape_1" + input: "gradients/DS_layer_type_1/concat_grad/ConcatOffset:1" + input: "gradients/DS_layer_type_1/concat_grad/ShapeN:1" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "gradients/DS_layer_type_0/Reshape_2_grad/Shape" + op: "Shape" + input: "DS_layer_type_0/add_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Reshape_2_grad/Reshape" + op: "Reshape" + input: "gradients/DS_layer_type_0/concat_grad/Slice" + input: "gradients/DS_layer_type_0/Reshape_2_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Reshape_5_grad/Shape" + op: "Shape" + input: "DS_layer_type_0/add_3" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Reshape_5_grad/Reshape" + op: "Reshape" + input: "gradients/DS_layer_type_0/concat_grad/Slice_1" + input: "gradients/DS_layer_type_0/Reshape_5_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Reshape_2_grad/Shape" + op: "Shape" + input: "DS_layer_type_1/add_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Reshape_2_grad/Reshape" + op: "Reshape" + input: "gradients/DS_layer_type_1/concat_grad/Slice" + input: "gradients/DS_layer_type_1/Reshape_2_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Reshape_5_grad/Shape" + op: "Shape" + input: "DS_layer_type_1/add_3" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Reshape_5_grad/Reshape" + op: "Reshape" + input: "gradients/DS_layer_type_1/concat_grad/Slice_1" + input: "gradients/DS_layer_type_1/Reshape_5_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/add_1_grad/Shape" + op: "Shape" + input: "DS_layer_type_0/Reshape_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/add_1_grad/Shape_1" + op: "Shape" + input: "DS_layer_type_0/Tanh" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/add_1_grad/BroadcastGradientArgs" + op: "BroadcastGradientArgs" + input: "gradients/DS_layer_type_0/add_1_grad/Shape" + input: "gradients/DS_layer_type_0/add_1_grad/Shape_1" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/add_1_grad/Sum" + op: "Sum" + input: "gradients/DS_layer_type_0/Reshape_2_grad/Reshape" + input: "gradients/DS_layer_type_0/add_1_grad/BroadcastGradientArgs" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_0/add_1_grad/Reshape" + op: "Reshape" + input: "gradients/DS_layer_type_0/add_1_grad/Sum" + input: "gradients/DS_layer_type_0/add_1_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/add_1_grad/Sum_1" + op: "Sum" + input: "gradients/DS_layer_type_0/Reshape_2_grad/Reshape" + input: "gradients/DS_layer_type_0/add_1_grad/BroadcastGradientArgs:1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_0/add_1_grad/Reshape_1" + op: "Reshape" + input: "gradients/DS_layer_type_0/add_1_grad/Sum_1" + input: "gradients/DS_layer_type_0/add_1_grad/Shape_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/add_3_grad/Shape" + op: "Shape" + input: "DS_layer_type_0/Reshape_4" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/add_3_grad/Shape_1" + op: "Shape" + input: "DS_layer_type_0/Tanh_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/add_3_grad/BroadcastGradientArgs" + op: "BroadcastGradientArgs" + input: "gradients/DS_layer_type_0/add_3_grad/Shape" + input: "gradients/DS_layer_type_0/add_3_grad/Shape_1" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/add_3_grad/Sum" + op: "Sum" + input: "gradients/DS_layer_type_0/Reshape_5_grad/Reshape" + input: "gradients/DS_layer_type_0/add_3_grad/BroadcastGradientArgs" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_0/add_3_grad/Reshape" + op: "Reshape" + input: "gradients/DS_layer_type_0/add_3_grad/Sum" + input: "gradients/DS_layer_type_0/add_3_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/add_3_grad/Sum_1" + op: "Sum" + input: "gradients/DS_layer_type_0/Reshape_5_grad/Reshape" + input: "gradients/DS_layer_type_0/add_3_grad/BroadcastGradientArgs:1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_0/add_3_grad/Reshape_1" + op: "Reshape" + input: "gradients/DS_layer_type_0/add_3_grad/Sum_1" + input: "gradients/DS_layer_type_0/add_3_grad/Shape_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/add_1_grad/Shape" + op: "Shape" + input: "DS_layer_type_1/Reshape_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/add_1_grad/Shape_1" + op: "Shape" + input: "DS_layer_type_1/Tanh" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/add_1_grad/BroadcastGradientArgs" + op: "BroadcastGradientArgs" + input: "gradients/DS_layer_type_1/add_1_grad/Shape" + input: "gradients/DS_layer_type_1/add_1_grad/Shape_1" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/add_1_grad/Sum" + op: "Sum" + input: "gradients/DS_layer_type_1/Reshape_2_grad/Reshape" + input: "gradients/DS_layer_type_1/add_1_grad/BroadcastGradientArgs" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_1/add_1_grad/Reshape" + op: "Reshape" + input: "gradients/DS_layer_type_1/add_1_grad/Sum" + input: "gradients/DS_layer_type_1/add_1_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/add_1_grad/Sum_1" + op: "Sum" + input: "gradients/DS_layer_type_1/Reshape_2_grad/Reshape" + input: "gradients/DS_layer_type_1/add_1_grad/BroadcastGradientArgs:1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_1/add_1_grad/Reshape_1" + op: "Reshape" + input: "gradients/DS_layer_type_1/add_1_grad/Sum_1" + input: "gradients/DS_layer_type_1/add_1_grad/Shape_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/add_3_grad/Shape" + op: "Shape" + input: "DS_layer_type_1/Reshape_4" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/add_3_grad/Shape_1" + op: "Shape" + input: "DS_layer_type_1/Tanh_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/add_3_grad/BroadcastGradientArgs" + op: "BroadcastGradientArgs" + input: "gradients/DS_layer_type_1/add_3_grad/Shape" + input: "gradients/DS_layer_type_1/add_3_grad/Shape_1" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/add_3_grad/Sum" + op: "Sum" + input: "gradients/DS_layer_type_1/Reshape_5_grad/Reshape" + input: "gradients/DS_layer_type_1/add_3_grad/BroadcastGradientArgs" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_1/add_3_grad/Reshape" + op: "Reshape" + input: "gradients/DS_layer_type_1/add_3_grad/Sum" + input: "gradients/DS_layer_type_1/add_3_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/add_3_grad/Sum_1" + op: "Sum" + input: "gradients/DS_layer_type_1/Reshape_5_grad/Reshape" + input: "gradients/DS_layer_type_1/add_3_grad/BroadcastGradientArgs:1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_1/add_3_grad/Reshape_1" + op: "Reshape" + input: "gradients/DS_layer_type_1/add_3_grad/Sum_1" + input: "gradients/DS_layer_type_1/add_3_grad/Shape_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Tanh_grad/TanhGrad" + op: "TanhGrad" + input: "DS_layer_type_0/Tanh" + input: "gradients/DS_layer_type_0/add_1_grad/Reshape_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "gradients/DS_layer_type_0/Tanh_1_grad/TanhGrad" + op: "TanhGrad" + input: "DS_layer_type_0/Tanh_1" + input: "gradients/DS_layer_type_0/add_3_grad/Reshape_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "gradients/DS_layer_type_1/Tanh_grad/TanhGrad" + op: "TanhGrad" + input: "DS_layer_type_1/Tanh" + input: "gradients/DS_layer_type_1/add_1_grad/Reshape_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "gradients/DS_layer_type_1/Tanh_1_grad/TanhGrad" + op: "TanhGrad" + input: "DS_layer_type_1/Tanh_1" + input: "gradients/DS_layer_type_1/add_3_grad/Reshape_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } +} +node { + name: "gradients/DS_layer_type_0/add_grad/Shape" + op: "Shape" + input: "DS_layer_type_0/MatMul" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/add_grad/Shape_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\001\000\000\000\001\000\000\000" + } + } + } +} +node { + name: "gradients/DS_layer_type_0/add_grad/BroadcastGradientArgs" + op: "BroadcastGradientArgs" + input: "gradients/DS_layer_type_0/add_grad/Shape" + input: "gradients/DS_layer_type_0/add_grad/Shape_1" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/add_grad/Sum" + op: "Sum" + input: "gradients/DS_layer_type_0/Tanh_grad/TanhGrad" + input: "gradients/DS_layer_type_0/add_grad/BroadcastGradientArgs" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_0/add_grad/Reshape" + op: "Reshape" + input: "gradients/DS_layer_type_0/add_grad/Sum" + input: "gradients/DS_layer_type_0/add_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/add_2_grad/Shape" + op: "Shape" + input: "DS_layer_type_0/MatMul_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/add_2_grad/Shape_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\001\000\000\000\001\000\000\000" + } + } + } +} +node { + name: "gradients/DS_layer_type_0/add_2_grad/BroadcastGradientArgs" + op: "BroadcastGradientArgs" + input: "gradients/DS_layer_type_0/add_2_grad/Shape" + input: "gradients/DS_layer_type_0/add_2_grad/Shape_1" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/add_2_grad/Sum" + op: "Sum" + input: "gradients/DS_layer_type_0/Tanh_1_grad/TanhGrad" + input: "gradients/DS_layer_type_0/add_2_grad/BroadcastGradientArgs" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_0/add_2_grad/Reshape" + op: "Reshape" + input: "gradients/DS_layer_type_0/add_2_grad/Sum" + input: "gradients/DS_layer_type_0/add_2_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/add_grad/Shape" + op: "Shape" + input: "DS_layer_type_1/MatMul" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/add_grad/Shape_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\001\000\000\000\001\000\000\000" + } + } + } +} +node { + name: "gradients/DS_layer_type_1/add_grad/BroadcastGradientArgs" + op: "BroadcastGradientArgs" + input: "gradients/DS_layer_type_1/add_grad/Shape" + input: "gradients/DS_layer_type_1/add_grad/Shape_1" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/add_grad/Sum" + op: "Sum" + input: "gradients/DS_layer_type_1/Tanh_grad/TanhGrad" + input: "gradients/DS_layer_type_1/add_grad/BroadcastGradientArgs" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_1/add_grad/Reshape" + op: "Reshape" + input: "gradients/DS_layer_type_1/add_grad/Sum" + input: "gradients/DS_layer_type_1/add_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/add_2_grad/Shape" + op: "Shape" + input: "DS_layer_type_1/MatMul_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/add_2_grad/Shape_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\001\000\000\000\001\000\000\000" + } + } + } +} +node { + name: "gradients/DS_layer_type_1/add_2_grad/BroadcastGradientArgs" + op: "BroadcastGradientArgs" + input: "gradients/DS_layer_type_1/add_2_grad/Shape" + input: "gradients/DS_layer_type_1/add_2_grad/Shape_1" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/add_2_grad/Sum" + op: "Sum" + input: "gradients/DS_layer_type_1/Tanh_1_grad/TanhGrad" + input: "gradients/DS_layer_type_1/add_2_grad/BroadcastGradientArgs" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } + attr { + key: "keep_dims" + value { + b: false + } + } +} +node { + name: "gradients/DS_layer_type_1/add_2_grad/Reshape" + op: "Reshape" + input: "gradients/DS_layer_type_1/add_2_grad/Sum" + input: "gradients/DS_layer_type_1/add_2_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_grad/MatMul" + op: "MatMul" + input: "gradients/DS_layer_type_0/add_grad/Reshape" + input: "DS_layer_type_0/matrix_1_0/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "transpose_a" + value { + b: false + } + } + attr { + key: "transpose_b" + value { + b: true + } + } +} +node { + name: "gradients/DS_layer_type_0/MatMul_1_grad/MatMul" + op: "MatMul" + input: "gradients/DS_layer_type_0/add_2_grad/Reshape" + input: "DS_layer_type_0/matrix_1_1/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "transpose_a" + value { + b: false + } + } + attr { + key: "transpose_b" + value { + b: true + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_grad/MatMul" + op: "MatMul" + input: "gradients/DS_layer_type_1/add_grad/Reshape" + input: "DS_layer_type_1/matrix_1_0/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "transpose_a" + value { + b: false + } + } + attr { + key: "transpose_b" + value { + b: true + } + } +} +node { + name: "gradients/DS_layer_type_1/MatMul_1_grad/MatMul" + op: "MatMul" + input: "gradients/DS_layer_type_1/add_2_grad/Reshape" + input: "DS_layer_type_1/matrix_1_1/read" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "transpose_a" + value { + b: false + } + } + attr { + key: "transpose_b" + value { + b: true + } + } +} +node { + name: "gradients/AddN_2" + op: "AddN" + input: "gradients/DS_layer_type_0/add_1_grad/Reshape" + input: "gradients/DS_layer_type_0/MatMul_grad/MatMul" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@gradients/DS_layer_type_0/add_1_grad/Reshape" + } + } + } +} +node { + name: "gradients/DS_layer_type_0/Reshape_1_grad/Shape" + op: "Shape" + input: "DS_layer_type_0/Slice_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Reshape_1_grad/Reshape" + op: "Reshape" + input: "gradients/AddN_2" + input: "gradients/DS_layer_type_0/Reshape_1_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/AddN_3" + op: "AddN" + input: "gradients/DS_layer_type_0/add_3_grad/Reshape" + input: "gradients/DS_layer_type_0/MatMul_1_grad/MatMul" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@gradients/DS_layer_type_0/add_3_grad/Reshape" + } + } + } +} +node { + name: "gradients/DS_layer_type_0/Reshape_4_grad/Shape" + op: "Shape" + input: "DS_layer_type_0/Slice_3" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Reshape_4_grad/Reshape" + op: "Reshape" + input: "gradients/AddN_3" + input: "gradients/DS_layer_type_0/Reshape_4_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/AddN_4" + op: "AddN" + input: "gradients/DS_layer_type_1/add_1_grad/Reshape" + input: "gradients/DS_layer_type_1/MatMul_grad/MatMul" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@gradients/DS_layer_type_1/add_1_grad/Reshape" + } + } + } +} +node { + name: "gradients/DS_layer_type_1/Reshape_1_grad/Shape" + op: "Shape" + input: "DS_layer_type_1/Slice_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Reshape_1_grad/Reshape" + op: "Reshape" + input: "gradients/AddN_4" + input: "gradients/DS_layer_type_1/Reshape_1_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/AddN_5" + op: "AddN" + input: "gradients/DS_layer_type_1/add_3_grad/Reshape" + input: "gradients/DS_layer_type_1/MatMul_1_grad/MatMul" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@gradients/DS_layer_type_1/add_3_grad/Reshape" + } + } + } +} +node { + name: "gradients/DS_layer_type_1/Reshape_4_grad/Shape" + op: "Shape" + input: "DS_layer_type_1/Slice_3" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Reshape_4_grad/Reshape" + op: "Reshape" + input: "gradients/AddN_5" + input: "gradients/DS_layer_type_1/Reshape_4_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_1_grad/Rank" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 2 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_1_grad/Shape" + op: "Shape" + input: "DS_layer_type_0/Slice_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_1_grad/stack/1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_1_grad/stack" + op: "Pack" + input: "gradients/DS_layer_type_0/Slice_1_grad/Rank" + input: "gradients/DS_layer_type_0/Slice_1_grad/stack/1" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "axis" + value { + i: 0 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_1_grad/Reshape" + op: "Reshape" + input: "DS_layer_type_0/Slice_1/begin" + input: "gradients/DS_layer_type_0/Slice_1_grad/stack" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_1_grad/Shape_1" + op: "Shape" + input: "DS_layer_type_0/Reshape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_1_grad/sub" + op: "Sub" + input: "gradients/DS_layer_type_0/Slice_1_grad/Shape_1" + input: "gradients/DS_layer_type_0/Slice_1_grad/Shape" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_1_grad/sub_1" + op: "Sub" + input: "gradients/DS_layer_type_0/Slice_1_grad/sub" + input: "DS_layer_type_0/Slice_1/begin" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_1_grad/Reshape_1" + op: "Reshape" + input: "gradients/DS_layer_type_0/Slice_1_grad/sub_1" + input: "gradients/DS_layer_type_0/Slice_1_grad/stack" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_1_grad/concat/axis" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_1_grad/concat" + op: "ConcatV2" + input: "gradients/DS_layer_type_0/Slice_1_grad/Reshape" + input: "gradients/DS_layer_type_0/Slice_1_grad/Reshape_1" + input: "gradients/DS_layer_type_0/Slice_1_grad/concat/axis" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_1_grad/Pad" + op: "Pad" + input: "gradients/DS_layer_type_0/Reshape_1_grad/Reshape" + input: "gradients/DS_layer_type_0/Slice_1_grad/concat" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tpaddings" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_3_grad/Rank" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 2 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_3_grad/Shape" + op: "Shape" + input: "DS_layer_type_0/Slice_3" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_3_grad/stack/1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_3_grad/stack" + op: "Pack" + input: "gradients/DS_layer_type_0/Slice_3_grad/Rank" + input: "gradients/DS_layer_type_0/Slice_3_grad/stack/1" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "axis" + value { + i: 0 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_3_grad/Reshape" + op: "Reshape" + input: "DS_layer_type_0/Slice_3/begin" + input: "gradients/DS_layer_type_0/Slice_3_grad/stack" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_3_grad/Shape_1" + op: "Shape" + input: "DS_layer_type_0/Reshape_3" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_3_grad/sub" + op: "Sub" + input: "gradients/DS_layer_type_0/Slice_3_grad/Shape_1" + input: "gradients/DS_layer_type_0/Slice_3_grad/Shape" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_3_grad/sub_1" + op: "Sub" + input: "gradients/DS_layer_type_0/Slice_3_grad/sub" + input: "DS_layer_type_0/Slice_3/begin" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_3_grad/Reshape_1" + op: "Reshape" + input: "gradients/DS_layer_type_0/Slice_3_grad/sub_1" + input: "gradients/DS_layer_type_0/Slice_3_grad/stack" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_3_grad/concat/axis" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_3_grad/concat" + op: "ConcatV2" + input: "gradients/DS_layer_type_0/Slice_3_grad/Reshape" + input: "gradients/DS_layer_type_0/Slice_3_grad/Reshape_1" + input: "gradients/DS_layer_type_0/Slice_3_grad/concat/axis" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_3_grad/Pad" + op: "Pad" + input: "gradients/DS_layer_type_0/Reshape_4_grad/Reshape" + input: "gradients/DS_layer_type_0/Slice_3_grad/concat" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tpaddings" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_1_grad/Rank" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 2 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_1_grad/Shape" + op: "Shape" + input: "DS_layer_type_1/Slice_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_1_grad/stack/1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_1_grad/stack" + op: "Pack" + input: "gradients/DS_layer_type_1/Slice_1_grad/Rank" + input: "gradients/DS_layer_type_1/Slice_1_grad/stack/1" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "axis" + value { + i: 0 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_1_grad/Reshape" + op: "Reshape" + input: "DS_layer_type_1/Slice_1/begin" + input: "gradients/DS_layer_type_1/Slice_1_grad/stack" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_1_grad/Shape_1" + op: "Shape" + input: "DS_layer_type_1/Reshape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_1_grad/sub" + op: "Sub" + input: "gradients/DS_layer_type_1/Slice_1_grad/Shape_1" + input: "gradients/DS_layer_type_1/Slice_1_grad/Shape" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_1_grad/sub_1" + op: "Sub" + input: "gradients/DS_layer_type_1/Slice_1_grad/sub" + input: "DS_layer_type_1/Slice_1/begin" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_1_grad/Reshape_1" + op: "Reshape" + input: "gradients/DS_layer_type_1/Slice_1_grad/sub_1" + input: "gradients/DS_layer_type_1/Slice_1_grad/stack" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_1_grad/concat/axis" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_1_grad/concat" + op: "ConcatV2" + input: "gradients/DS_layer_type_1/Slice_1_grad/Reshape" + input: "gradients/DS_layer_type_1/Slice_1_grad/Reshape_1" + input: "gradients/DS_layer_type_1/Slice_1_grad/concat/axis" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_1_grad/Pad" + op: "Pad" + input: "gradients/DS_layer_type_1/Reshape_1_grad/Reshape" + input: "gradients/DS_layer_type_1/Slice_1_grad/concat" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tpaddings" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_3_grad/Rank" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 2 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_3_grad/Shape" + op: "Shape" + input: "DS_layer_type_1/Slice_3" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_3_grad/stack/1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_3_grad/stack" + op: "Pack" + input: "gradients/DS_layer_type_1/Slice_3_grad/Rank" + input: "gradients/DS_layer_type_1/Slice_3_grad/stack/1" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "axis" + value { + i: 0 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_3_grad/Reshape" + op: "Reshape" + input: "DS_layer_type_1/Slice_3/begin" + input: "gradients/DS_layer_type_1/Slice_3_grad/stack" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_3_grad/Shape_1" + op: "Shape" + input: "DS_layer_type_1/Reshape_3" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_3_grad/sub" + op: "Sub" + input: "gradients/DS_layer_type_1/Slice_3_grad/Shape_1" + input: "gradients/DS_layer_type_1/Slice_3_grad/Shape" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_3_grad/sub_1" + op: "Sub" + input: "gradients/DS_layer_type_1/Slice_3_grad/sub" + input: "DS_layer_type_1/Slice_3/begin" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_3_grad/Reshape_1" + op: "Reshape" + input: "gradients/DS_layer_type_1/Slice_3_grad/sub_1" + input: "gradients/DS_layer_type_1/Slice_3_grad/stack" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_3_grad/concat/axis" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_3_grad/concat" + op: "ConcatV2" + input: "gradients/DS_layer_type_1/Slice_3_grad/Reshape" + input: "gradients/DS_layer_type_1/Slice_3_grad/Reshape_1" + input: "gradients/DS_layer_type_1/Slice_3_grad/concat/axis" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_3_grad/Pad" + op: "Pad" + input: "gradients/DS_layer_type_1/Reshape_4_grad/Reshape" + input: "gradients/DS_layer_type_1/Slice_3_grad/concat" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tpaddings" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Reshape_grad/Shape" + op: "Shape" + input: "DS_layer_type_0/Slice" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Reshape_grad/Reshape" + op: "Reshape" + input: "gradients/DS_layer_type_0/Slice_1_grad/Pad" + input: "gradients/DS_layer_type_0/Reshape_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Reshape_3_grad/Shape" + op: "Shape" + input: "DS_layer_type_0/Slice_2" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Reshape_3_grad/Reshape" + op: "Reshape" + input: "gradients/DS_layer_type_0/Slice_3_grad/Pad" + input: "gradients/DS_layer_type_0/Reshape_3_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Reshape_grad/Shape" + op: "Shape" + input: "DS_layer_type_1/Slice" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Reshape_grad/Reshape" + op: "Reshape" + input: "gradients/DS_layer_type_1/Slice_1_grad/Pad" + input: "gradients/DS_layer_type_1/Reshape_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Reshape_3_grad/Shape" + op: "Shape" + input: "DS_layer_type_1/Slice_2" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Reshape_3_grad/Reshape" + op: "Reshape" + input: "gradients/DS_layer_type_1/Slice_3_grad/Pad" + input: "gradients/DS_layer_type_1/Reshape_3_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_grad/Rank" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 2 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_grad/Shape" + op: "Shape" + input: "DS_layer_type_0/Slice" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_grad/stack/1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_grad/stack" + op: "Pack" + input: "gradients/DS_layer_type_0/Slice_grad/Rank" + input: "gradients/DS_layer_type_0/Slice_grad/stack/1" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "axis" + value { + i: 0 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_grad/Reshape" + op: "Reshape" + input: "DS_layer_type_0/Slice/begin" + input: "gradients/DS_layer_type_0/Slice_grad/stack" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_grad/Shape_1" + op: "Shape" + input: "Reshape_4" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_grad/sub" + op: "Sub" + input: "gradients/DS_layer_type_0/Slice_grad/Shape_1" + input: "gradients/DS_layer_type_0/Slice_grad/Shape" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_grad/sub_1" + op: "Sub" + input: "gradients/DS_layer_type_0/Slice_grad/sub" + input: "DS_layer_type_0/Slice/begin" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_grad/Reshape_1" + op: "Reshape" + input: "gradients/DS_layer_type_0/Slice_grad/sub_1" + input: "gradients/DS_layer_type_0/Slice_grad/stack" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_grad/concat/axis" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_grad/concat" + op: "ConcatV2" + input: "gradients/DS_layer_type_0/Slice_grad/Reshape" + input: "gradients/DS_layer_type_0/Slice_grad/Reshape_1" + input: "gradients/DS_layer_type_0/Slice_grad/concat/axis" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_grad/Pad" + op: "Pad" + input: "gradients/DS_layer_type_0/Reshape_grad/Reshape" + input: "gradients/DS_layer_type_0/Slice_grad/concat" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tpaddings" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_2_grad/Rank" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 2 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_2_grad/Shape" + op: "Shape" + input: "DS_layer_type_0/Slice_2" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_2_grad/stack/1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_2_grad/stack" + op: "Pack" + input: "gradients/DS_layer_type_0/Slice_2_grad/Rank" + input: "gradients/DS_layer_type_0/Slice_2_grad/stack/1" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "axis" + value { + i: 0 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_2_grad/Reshape" + op: "Reshape" + input: "DS_layer_type_0/Slice_2/begin" + input: "gradients/DS_layer_type_0/Slice_2_grad/stack" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_2_grad/Shape_1" + op: "Shape" + input: "Reshape_4" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_2_grad/sub" + op: "Sub" + input: "gradients/DS_layer_type_0/Slice_2_grad/Shape_1" + input: "gradients/DS_layer_type_0/Slice_2_grad/Shape" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_2_grad/sub_1" + op: "Sub" + input: "gradients/DS_layer_type_0/Slice_2_grad/sub" + input: "DS_layer_type_0/Slice_2/begin" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_2_grad/Reshape_1" + op: "Reshape" + input: "gradients/DS_layer_type_0/Slice_2_grad/sub_1" + input: "gradients/DS_layer_type_0/Slice_2_grad/stack" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_2_grad/concat/axis" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_2_grad/concat" + op: "ConcatV2" + input: "gradients/DS_layer_type_0/Slice_2_grad/Reshape" + input: "gradients/DS_layer_type_0/Slice_2_grad/Reshape_1" + input: "gradients/DS_layer_type_0/Slice_2_grad/concat/axis" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_0/Slice_2_grad/Pad" + op: "Pad" + input: "gradients/DS_layer_type_0/Reshape_3_grad/Reshape" + input: "gradients/DS_layer_type_0/Slice_2_grad/concat" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tpaddings" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_grad/Rank" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 2 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_grad/Shape" + op: "Shape" + input: "DS_layer_type_1/Slice" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_grad/stack/1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_grad/stack" + op: "Pack" + input: "gradients/DS_layer_type_1/Slice_grad/Rank" + input: "gradients/DS_layer_type_1/Slice_grad/stack/1" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "axis" + value { + i: 0 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_grad/Reshape" + op: "Reshape" + input: "DS_layer_type_1/Slice/begin" + input: "gradients/DS_layer_type_1/Slice_grad/stack" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_grad/Shape_1" + op: "Shape" + input: "Reshape_6" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_grad/sub" + op: "Sub" + input: "gradients/DS_layer_type_1/Slice_grad/Shape_1" + input: "gradients/DS_layer_type_1/Slice_grad/Shape" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_grad/sub_1" + op: "Sub" + input: "gradients/DS_layer_type_1/Slice_grad/sub" + input: "DS_layer_type_1/Slice/begin" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_grad/Reshape_1" + op: "Reshape" + input: "gradients/DS_layer_type_1/Slice_grad/sub_1" + input: "gradients/DS_layer_type_1/Slice_grad/stack" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_grad/concat/axis" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_grad/concat" + op: "ConcatV2" + input: "gradients/DS_layer_type_1/Slice_grad/Reshape" + input: "gradients/DS_layer_type_1/Slice_grad/Reshape_1" + input: "gradients/DS_layer_type_1/Slice_grad/concat/axis" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_grad/Pad" + op: "Pad" + input: "gradients/DS_layer_type_1/Reshape_grad/Reshape" + input: "gradients/DS_layer_type_1/Slice_grad/concat" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tpaddings" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_2_grad/Rank" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 2 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_2_grad/Shape" + op: "Shape" + input: "DS_layer_type_1/Slice_2" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_2_grad/stack/1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_2_grad/stack" + op: "Pack" + input: "gradients/DS_layer_type_1/Slice_2_grad/Rank" + input: "gradients/DS_layer_type_1/Slice_2_grad/stack/1" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "axis" + value { + i: 0 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_2_grad/Reshape" + op: "Reshape" + input: "DS_layer_type_1/Slice_2/begin" + input: "gradients/DS_layer_type_1/Slice_2_grad/stack" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_2_grad/Shape_1" + op: "Shape" + input: "Reshape_6" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_2_grad/sub" + op: "Sub" + input: "gradients/DS_layer_type_1/Slice_2_grad/Shape_1" + input: "gradients/DS_layer_type_1/Slice_2_grad/Shape" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_2_grad/sub_1" + op: "Sub" + input: "gradients/DS_layer_type_1/Slice_2_grad/sub" + input: "DS_layer_type_1/Slice_2/begin" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_2_grad/Reshape_1" + op: "Reshape" + input: "gradients/DS_layer_type_1/Slice_2_grad/sub_1" + input: "gradients/DS_layer_type_1/Slice_2_grad/stack" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_2_grad/concat/axis" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_2_grad/concat" + op: "ConcatV2" + input: "gradients/DS_layer_type_1/Slice_2_grad/Reshape" + input: "gradients/DS_layer_type_1/Slice_2_grad/Reshape_1" + input: "gradients/DS_layer_type_1/Slice_2_grad/concat/axis" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/DS_layer_type_1/Slice_2_grad/Pad" + op: "Pad" + input: "gradients/DS_layer_type_1/Reshape_3_grad/Reshape" + input: "gradients/DS_layer_type_1/Slice_2_grad/concat" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tpaddings" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/AddN_6" + op: "AddN" + input: "gradients/DS_layer_type_0/Reshape_6_grad/Reshape" + input: "gradients/DS_layer_type_0/Slice_grad/Pad" + input: "gradients/DS_layer_type_0/Slice_2_grad/Pad" + attr { + key: "N" + value { + i: 3 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@gradients/DS_layer_type_0/Reshape_6_grad/Reshape" + } + } + } +} +node { + name: "gradients/Reshape_4_grad/Shape" + op: "Shape" + input: "Slice" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/Reshape_4_grad/Reshape" + op: "Reshape" + input: "gradients/AddN_6" + input: "gradients/Reshape_4_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/AddN_7" + op: "AddN" + input: "gradients/DS_layer_type_1/Reshape_6_grad/Reshape" + input: "gradients/DS_layer_type_1/Slice_grad/Pad" + input: "gradients/DS_layer_type_1/Slice_2_grad/Pad" + attr { + key: "N" + value { + i: 3 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@gradients/DS_layer_type_1/Reshape_6_grad/Reshape" + } + } + } +} +node { + name: "gradients/Reshape_6_grad/Shape" + op: "Shape" + input: "Slice_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/Reshape_6_grad/Reshape" + op: "Reshape" + input: "gradients/AddN_7" + input: "gradients/Reshape_6_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/Slice_grad/Rank" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 2 + } + } + } +} +node { + name: "gradients/Slice_grad/Shape" + op: "Shape" + input: "Slice" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/Slice_grad/stack/1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "gradients/Slice_grad/stack" + op: "Pack" + input: "gradients/Slice_grad/Rank" + input: "gradients/Slice_grad/stack/1" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "axis" + value { + i: 0 + } + } +} +node { + name: "gradients/Slice_grad/Reshape" + op: "Reshape" + input: "Slice/begin" + input: "gradients/Slice_grad/stack" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/Slice_grad/Shape_1" + op: "Shape" + input: "Reshape_3" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/Slice_grad/sub" + op: "Sub" + input: "gradients/Slice_grad/Shape_1" + input: "gradients/Slice_grad/Shape" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/Slice_grad/sub_1" + op: "Sub" + input: "gradients/Slice_grad/sub" + input: "Slice/begin" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/Slice_grad/Reshape_1" + op: "Reshape" + input: "gradients/Slice_grad/sub_1" + input: "gradients/Slice_grad/stack" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/Slice_grad/concat/axis" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "gradients/Slice_grad/concat" + op: "ConcatV2" + input: "gradients/Slice_grad/Reshape" + input: "gradients/Slice_grad/Reshape_1" + input: "gradients/Slice_grad/concat/axis" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/Slice_grad/Pad" + op: "Pad" + input: "gradients/Reshape_4_grad/Reshape" + input: "gradients/Slice_grad/concat" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tpaddings" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/Slice_1_grad/Rank" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 2 + } + } + } +} +node { + name: "gradients/Slice_1_grad/Shape" + op: "Shape" + input: "Slice_1" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/Slice_1_grad/stack/1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "gradients/Slice_1_grad/stack" + op: "Pack" + input: "gradients/Slice_1_grad/Rank" + input: "gradients/Slice_1_grad/stack/1" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "axis" + value { + i: 0 + } + } +} +node { + name: "gradients/Slice_1_grad/Reshape" + op: "Reshape" + input: "Slice_1/begin" + input: "gradients/Slice_1_grad/stack" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/Slice_1_grad/Shape_1" + op: "Shape" + input: "Reshape_3" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/Slice_1_grad/sub" + op: "Sub" + input: "gradients/Slice_1_grad/Shape_1" + input: "gradients/Slice_1_grad/Shape" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/Slice_1_grad/sub_1" + op: "Sub" + input: "gradients/Slice_1_grad/sub" + input: "Slice_1/begin" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/Slice_1_grad/Reshape_1" + op: "Reshape" + input: "gradients/Slice_1_grad/sub_1" + input: "gradients/Slice_1_grad/stack" + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/Slice_1_grad/concat/axis" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 1 + } + } + } +} +node { + name: "gradients/Slice_1_grad/concat" + op: "ConcatV2" + input: "gradients/Slice_1_grad/Reshape" + input: "gradients/Slice_1_grad/Reshape_1" + input: "gradients/Slice_1_grad/concat/axis" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "Tidx" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/Slice_1_grad/Pad" + op: "Pad" + input: "gradients/Reshape_6_grad/Reshape" + input: "gradients/Slice_1_grad/concat" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tpaddings" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/AddN_8" + op: "AddN" + input: "gradients/Slice_grad/Pad" + input: "gradients/Slice_1_grad/Pad" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "_class" + value { + list { + s: "loc:@gradients/Slice_grad/Pad" + } + } + } +} +node { + name: "gradients/Reshape_3_grad/Shape" + op: "Shape" + input: "Reshape_2" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "gradients/Reshape_3_grad/Reshape" + op: "Reshape" + input: "gradients/AddN_8" + input: "gradients/Reshape_3_grad/Shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "strided_slice_10/stack" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 0 + } + } + } +} +node { + name: "strided_slice_10/stack_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "strided_slice_10/stack_2" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "strided_slice_10" + op: "StridedSlice" + input: "t_natoms" + input: "strided_slice_10/stack" + input: "strided_slice_10/stack_1" + input: "strided_slice_10/stack_2" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "begin_mask" + value { + i: 0 + } + } + attr { + key: "ellipsis_mask" + value { + i: 0 + } + } + attr { + key: "end_mask" + value { + i: 0 + } + } + attr { + key: "new_axis_mask" + value { + i: 0 + } + } + attr { + key: "shrink_axis_mask" + value { + i: 1 + } + } +} +node { + name: "mul_5/y" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 8 + } + } + } +} +node { + name: "mul_5" + op: "Mul" + input: "strided_slice_10" + input: "mul_5/y" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "Reshape_9/shape/0" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: -1 + } + } + } +} +node { + name: "Reshape_9/shape" + op: "Pack" + input: "Reshape_9/shape/0" + input: "mul_5" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "axis" + value { + i: 0 + } + } +} +node { + name: "Reshape_9" + op: "Reshape" + input: "gradients/Reshape_3_grad/Reshape" + input: "Reshape_9/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "ProdForceNorot" + op: "ProdForceNorot" + input: "Reshape_9" + input: "DescrptNorot:1" + input: "DescrptNorot:3" + input: "t_natoms" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "n_a_sel" + value { + i: 2 + } + } + attr { + key: "n_r_sel" + value { + i: 0 + } + } +} +node { + name: "strided_slice_11/stack" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "strided_slice_11/stack_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 2 + } + } + } +} +node { + name: "strided_slice_11/stack_2" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "strided_slice_11" + op: "StridedSlice" + input: "t_natoms" + input: "strided_slice_11/stack" + input: "strided_slice_11/stack_1" + input: "strided_slice_11/stack_2" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "begin_mask" + value { + i: 0 + } + } + attr { + key: "ellipsis_mask" + value { + i: 0 + } + } + attr { + key: "end_mask" + value { + i: 0 + } + } + attr { + key: "new_axis_mask" + value { + i: 0 + } + } + attr { + key: "shrink_axis_mask" + value { + i: 1 + } + } +} +node { + name: "mul_6/x" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 3 + } + } + } +} +node { + name: "mul_6" + op: "Mul" + input: "mul_6/x" + input: "strided_slice_11" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "force_test/shape/0" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: -1 + } + } + } +} +node { + name: "force_test/shape" + op: "Pack" + input: "force_test/shape/0" + input: "mul_6" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "axis" + value { + i: 0 + } + } +} +node { + name: "force_test" + op: "Reshape" + input: "ProdForceNorot" + input: "force_test/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "ProdVirialNorot" + op: "ProdVirialNorot" + input: "Reshape_9" + input: "DescrptNorot:1" + input: "DescrptNorot:2" + input: "DescrptNorot:3" + input: "t_natoms" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "n_a_sel" + value { + i: 2 + } + } + attr { + key: "n_r_sel" + value { + i: 0 + } + } +} +node { + name: "virial_test/shape" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + tensor_content: "\377\377\377\377\t\000\000\000" + } + } + } +} +node { + name: "virial_test" + op: "Reshape" + input: "ProdVirialNorot" + input: "virial_test/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "strided_slice_12/stack" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "strided_slice_12/stack_1" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 2 + } + } + } +} +node { + name: "strided_slice_12/stack_2" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 1 + } + } + int_val: 1 + } + } + } +} +node { + name: "strided_slice_12" + op: "StridedSlice" + input: "t_natoms" + input: "strided_slice_12/stack" + input: "strided_slice_12/stack_1" + input: "strided_slice_12/stack_2" + attr { + key: "Index" + value { + type: DT_INT32 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "begin_mask" + value { + i: 0 + } + } + attr { + key: "ellipsis_mask" + value { + i: 0 + } + } + attr { + key: "end_mask" + value { + i: 0 + } + } + attr { + key: "new_axis_mask" + value { + i: 0 + } + } + attr { + key: "shrink_axis_mask" + value { + i: 1 + } + } +} +node { + name: "mul_7/x" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: 9 + } + } + } +} +node { + name: "mul_7" + op: "Mul" + input: "mul_7/x" + input: "strided_slice_12" + attr { + key: "T" + value { + type: DT_INT32 + } + } +} +node { + name: "atom_virial_test/shape/0" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + } + int_val: -1 + } + } + } +} +node { + name: "atom_virial_test/shape" + op: "Pack" + input: "atom_virial_test/shape/0" + input: "mul_7" + attr { + key: "N" + value { + i: 2 + } + } + attr { + key: "T" + value { + type: DT_INT32 + } + } + attr { + key: "axis" + value { + i: 0 + } + } +} +node { + name: "atom_virial_test" + op: "Reshape" + input: "ProdVirialNorot:1" + input: "atom_virial_test/shape" + attr { + key: "T" + value { + type: DT_DOUBLE + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +library { +} diff --git a/source/tests/test_deeppot_a.py b/source/tests/test_deeppot_a.py index da1c19587f..73e9f1d7e6 100644 --- a/source/tests/test_deeppot_a.py +++ b/source/tests/test_deeppot_a.py @@ -5,7 +5,7 @@ from infer.convert2pb import convert_pbtxt_to_pb from deepmd.infer import DeepPot from deepmd.env import MODEL_VERSION -from common import tests_path +from common import tests_path, run_dp from deepmd.env import GLOBAL_NP_FLOAT_PRECISION if GLOBAL_NP_FLOAT_PRECISION == np.float32 : @@ -322,3 +322,25 @@ def test_ase(self): np.testing.assert_almost_equal(ff.ravel(), self.expected_f.ravel(), default_places) expected_se = np.sum(self.expected_e.reshape([nframes, -1]), axis = 1) np.testing.assert_almost_equal(ee.ravel(), expected_se.ravel(), default_places) + + +class TestModelConvert(unittest.TestCase): + def setUp(self): + self.coords = np.array([12.83, 2.56, 2.18, + 12.09, 2.87, 2.74, + 00.25, 3.32, 1.68, + 3.36, 3.00, 1.81, + 3.51, 2.51, 2.60, + 4.27, 3.22, 1.56]) + self.atype = [0, 1, 1, 0, 1, 1] + self.box = np.array([13., 0., 0., 0., 13., 0., 0., 0., 13.]) + + def test_convert_012(self): + old_model = "deeppot.pb" + new_model = "deeppot.pbtxt" + convert_pbtxt_to_pb(str(tests_path / "infer" / "sea_012.pbtxt"), old_model) + run_dp(f"dp convert-from 0.12 -i {old_model} -o {new_model}") + dp = DeepPot(new_model) + _, _, _, _, _ = dp.eval(self.coords, self.box, self.atype, atomic=True) + os.remove(old_model) + os.remove(new_model)