diff --git a/mindinsight/common/hook/debugger.py b/mindinsight/common/hook/debugger.py index 51d4f8c5..2140bcda 100644 --- a/mindinsight/common/hook/debugger.py +++ b/mindinsight/common/hook/debugger.py @@ -20,11 +20,11 @@ from mindinsight.conf import settings from mindinsight.utils.hook import BaseHook -def str2bool(string): +def enable_debugger_string(string): """Convert str to bool""" - if string.lower() == 'false': + if string.lower() in ('false', '0'): return False - if string.lower() == 'true': + if string.lower() in ('true', '1'): return True raise ValueError @@ -83,11 +83,11 @@ class Hook(BaseHook): """ parser.add_argument( '--enable-debugger', - type=str2bool, + type=enable_debugger_string, action=EnableDebuggerAction, default=False, help=""" - Enable debugger or not. + Enable debugger or not. The value can be True/False/1/0 (case insensitive). Default is False.""") parser.add_argument( diff --git a/mindinsight/debugger/debugger_grpc_server.py b/mindinsight/debugger/debugger_grpc_server.py index dcf8b10f..4133021d 100644 --- a/mindinsight/debugger/debugger_grpc_server.py +++ b/mindinsight/debugger/debugger_grpc_server.py @@ -504,9 +504,9 @@ class DebuggerGrpcServer(grpc_server_base.EventListenerServicer): } hit_params = {} for param in watchpoint_hit_proto.watch_condition.params: - if param.actual_value is not None and param.name not in \ - (ParamNameEnum.RTOL.value, ParamNameEnum.RANGE_START_INCLUSIVE.value, - ParamNameEnum.RANGE_END_INCLUSIVE.value): + if param.name not in (ParamNameEnum.RTOL.value, ParamNameEnum.RANGE_START_INCLUSIVE.value, + ParamNameEnum.RANGE_END_INCLUSIVE.value) \ + and watchpoint_hit_proto.error_code == 0: hit_params[param.name] = param.actual_value for i, param in enumerate(watchpoint_hit['watchpoint'].condition['params']): name = param['name'] @@ -514,8 +514,7 @@ class DebuggerGrpcServer(grpc_server_base.EventListenerServicer): watchpoint_hit['watchpoint'].condition['params'][i]['actual_value'] = hit_params[name] else: watchpoint_hit['watchpoint'].condition['params'][i]['actual_value'] = None - if watchpoint_hit_proto.error_code is not None: - watchpoint_hit['error_code'] = watchpoint_hit_proto.error_code + watchpoint_hit['error_code'] = watchpoint_hit_proto.error_code watchpoint_hits.append(watchpoint_hit) self._received_hit = watchpoint_hits reply = get_ack_reply() diff --git a/mindinsight/debugger/stream_operator/watchpoint_operator.py b/mindinsight/debugger/stream_operator/watchpoint_operator.py index ad4c1dcf..63157c3c 100644 --- a/mindinsight/debugger/stream_operator/watchpoint_operator.py +++ b/mindinsight/debugger/stream_operator/watchpoint_operator.py @@ -21,7 +21,7 @@ from mindinsight.debugger.common.exceptions.exceptions import DebuggerParamValue from mindinsight.debugger.common.log import LOGGER as log from mindinsight.debugger.common.utils import ServerStatus, \ Streams, is_cst_type -from mindinsight.debugger.conditionmgr.condition import ConditionIdEnum, TargetTypeEnum, ConditionContext +from mindinsight.debugger.conditionmgr.condition import ConditionIdEnum, ConditionContext from mindinsight.debugger.conditionmgr.recommender import get_basic_node_info from mindinsight.debugger.stream_handler.watchpoint_handler import validate_watch_condition @@ -84,10 +84,7 @@ class WatchpointOperator: raise DebuggerConditionUnavailableError( "Failed to create watchpoint as the condition is not available.") - if condition.supported_target_type in [TargetTypeEnum.ACTIVATION, TargetTypeEnum.GRADIENT, - TargetTypeEnum.WEIGHT]: - watch_nodes = get_basic_node_info(condition.supported_target_type.value, self._graph_stream).copy() - + watch_nodes = get_basic_node_info(condition.supported_target_type.value, self._graph_stream).copy() watchpoint_stream = self._watchpoint_stream watch_point_id = watchpoint_stream.create_watchpoint( self._condition_mgr, watch_condition, watch_nodes, params.get('watch_point_id')) diff --git a/tests/st/func/debugger/expect_results/restful_results/retrieve_single_watchpoint_hit.json b/tests/st/func/debugger/expect_results/restful_results/retrieve_single_watchpoint_hit.json index 83654d91..ee049013 100644 --- a/tests/st/func/debugger/expect_results/restful_results/retrieve_single_watchpoint_hit.json +++ b/tests/st/func/debugger/expect_results/restful_results/retrieve_single_watchpoint_hit.json @@ -1,1737 +1 @@ -{ - "tensor_history": [ - { - "name": "Default/TransData-op99:0", - "full_name": "Default/TransData-op99:0", - "node_type": "TransData", - "type": "output", - "graph_name": "graph_0", - "is_hit": true - }, - { - "name": "Default/args0:0", - "full_name": "Default/args0:0", - "node_type": "Parameter", - "type": "input", - "graph_name": "graph_0", - "is_hit": false - } - ], - "metadata": { - "state": "waiting", - "step": 2 - }, - "graph": { - "nodes": [ - { - "name": "Default", - "type": "name_scope", - "attr": {}, - "input": { - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradBiasAdd/BiasAddGrad-op21": { - "shape": [ - [ - 10 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradMatMul/MatMul[6]_5/MatMul-op24": { - "shape": [ - [ - 10, - 84 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradBiasAdd/BiasAddGrad-op29": { - "shape": [ - [ - 84 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradMatMul/MatMul[6]_5/MatMul-op32": { - "shape": [ - [ - 84, - 120 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradBiasAdd/BiasAddGrad-op37": { - "shape": [ - [ - 120 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradMatMul/MatMul[6]_5/MatMul-op40": { - "shape": [ - [ - 120, - 400 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/conv2-Conv2d/gradConv2D/Conv2DBackpropFilter-op48": { - "shape": [ - [ - 16, - 6, - 5, - 5 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/conv1-Conv2d/gradConv2D/Conv2DBackpropFilter-op55": { - "shape": [ - [ - 6, - 1, - 5, - 5 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - } - }, - "output": { - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/conv2-Conv2d/gradConv2D/Conv2DBackpropInput-op52": { - "shape": [ - [ - 16, - 6, - 5, - 5 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/conv1-Conv2d/gradConv2D/Conv2DBackpropFilter-op55": { - "shape": [ - [ - 32, - 1, - 32, - 32 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/gradMaxPoolWithArgmax/MaxPoolGradWithArgmax-op53": { - "shape": [ - [ - 32, - 6, - 4, - 14 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_UINT16]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/gradMaxPoolWithArgmax/MaxPoolGradWithArgmax-op46": { - "shape": [ - [ - 32, - 16, - 4, - 3 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_UINT16]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradMatMul/MatMul[6]_5/MatMul-op40": { - "shape": [ - [ - 32, - 400 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/gradReLU/ReluGrad-op36": { - "shape": [ - [ - 32, - 120 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradMatMul/MatMul[6]_5/MatMul-op32": { - "shape": [ - [ - 32, - 120 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/gradReLU/ReluGrad-op28": { - "shape": [ - [ - 32, - 84 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradMatMul/MatMul[6]_5/MatMul-op24": { - "shape": [ - [ - 32, - 84 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/conv2-Conv2d/gradConv2D/Conv2DBackpropFilter-op48": { - "shape": [ - [ - 32, - 6, - 14, - 14 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_loss_fn-SoftmaxCrossEntropyWithLogits/gradSoftmaxCrossEntropyWithLogits/Mul-op20": { - "shape": [ - [ - 32, - 10 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/gradReLU/ReluGradV2-op92": { - "shape": [ - [ - 32, - 1, - 10, - 10, - 2 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_UINT8]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/gradReLU/ReluGradV2-op94": { - "shape": [ - [ - 32, - 1, - 28, - 28, - 2 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_UINT8]" - } - }, - "output_i": 0, - "proxy_input": {}, - "proxy_output": {}, - "subnode_count": 7, - "independent_layout": false, - "watched": 1 - }, - { - "name": "Gradients", - "type": "name_scope", - "attr": {}, - "input": { - "Default/tuple_getitem[10]_0/tuple_getitem-op210": { - "shape": [ - [ - 32, - 10 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/ReLU-op15": { - "shape": [ - [ - 32, - 84 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/ReLU-op12": { - "shape": [ - [ - 32, - 120 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/Cast-op205": { - "shape": [ - [ - 32, - 16, - 10, - 10 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/tuple_getitem[10]_0/tuple_getitem-op206": { - "shape": [ - [ - 32, - 16, - 4, - 3 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_UINT16]" - }, - "Default/tuple_getitem[10]_0/tuple_getitem-op202": { - "shape": [ - [ - 32, - 1, - 10, - 10, - 2 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_UINT8]" - }, - "Default/tuple_getitem[10]_0/tuple_getitem-op197": { - "shape": [ - [ - 32, - 6, - 14, - 14 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/network-WithLossCell/_backbone-LeNet5/conv2-Conv2d/Cast-op188": { - "shape": [ - [ - 16, - 6, - 5, - 5 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/Cast-op195": { - "shape": [ - [ - 32, - 6, - 28, - 28 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/tuple_getitem[10]_0/tuple_getitem-op196": { - "shape": [ - [ - 32, - 6, - 4, - 14 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_UINT16]" - }, - "Default/tuple_getitem[10]_0/tuple_getitem-op192": { - "shape": [ - [ - 32, - 1, - 28, - 28, - 2 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_UINT8]" - }, - "Default/network-WithLossCell/_backbone-LeNet5/conv1-Conv2d/Cast-op190": { - "shape": [ - [ - 32, - 1, - 32, - 32 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/network-WithLossCell/_backbone-LeNet5/flatten-Flatten/Reshape-op9": { - "shape": [ - [ - 32, - 400 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - } - }, - "output": { - "Default/optimizer-Momentum/ApplyMomentum[8]_1/ApplyMomentum-op22": { - "shape": [ - [ - 10 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/optimizer-Momentum/ApplyMomentum[8]_1/ApplyMomentum-op30": { - "shape": [ - [ - 84 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/optimizer-Momentum/ApplyMomentum[8]_1/ApplyMomentum-op38": { - "shape": [ - [ - 120 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/optimizer-Momentum/ApplyMomentum[8]_1/ApplyMomentum-op49": { - "shape": [ - [ - 16, - 6, - 5, - 5 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/optimizer-Momentum/ApplyMomentum[8]_1/ApplyMomentum-op56": { - "shape": [ - [ - 6, - 1, - 5, - 5 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/optimizer-Momentum/ApplyMomentum[8]_1/ApplyMomentum-op25": { - "shape": [ - [ - 10, - 84 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/optimizer-Momentum/ApplyMomentum[8]_1/ApplyMomentum-op33": { - "shape": [ - [ - 84, - 120 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/optimizer-Momentum/ApplyMomentum[8]_1/ApplyMomentum-op41": { - "shape": [ - [ - 120, - 400 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - } - }, - "output_i": 0, - "proxy_input": {}, - "proxy_output": {}, - "subnode_count": 1, - "independent_layout": false, - "watched": 0 - } - ], - "scope_name": "", - "children": { - "nodes": [ - { - "name": "Default/TransData-op99", - "type": "TransData", - "attr": { - "dst_format": "dtype: DT_STRING\nstr_val: \"NC1HWC0\"\n", - "IsFeatureMapInputList": "dtype: DT_TUPLE\nvalues {\n dtype: DT_UINT64\n uint_val: 0\n}\n", - "pri_format": "dtype: DT_STRING\nstr_val: \"NC1HWC0\"\n", - "_datadump_original_names": "dtype: DT_TUPLE\n", - "IsFeatureMapOutput": "dtype: DT_BOOL\nbool_val: true\n", - "src_format": "dtype: DT_STRING\nstr_val: \"NCHW\"\n" - }, - "input": { - "Default/args0": { - "shape": [ - [ - 32, - 1, - 32, - 32 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - } - }, - "output": { - "Default/network-WithLossCell/_backbone-LeNet5/conv1-Conv2d/Cast-op190": { - "shape": [ - [ - 32, - 1, - 32, - 32 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - } - }, - "output_i": 0, - "proxy_input": {}, - "proxy_output": {}, - "subnode_count": 0, - "independent_layout": false, - "watched": 2 - }, - { - "name": "Default/Depend-op61", - "type": "Depend", - "attr": {}, - "input": { - "Default/network-WithLossCell/_loss_fn-SoftmaxCrossEntropyWithLogits/ReduceMean-op60": { - "shape": [ - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/optimizer-Momentum/make_tuple-op58": { - "shape": [ - [], - [], - [], - [], - [], - [], - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_BOOL', 'DT_BOOL', 'DT_BOOL', 'DT_BOOL', 'DT_BOOL', 'DT_BOOL', 'DT_BOOL', 'DT_BOOL']" - } - }, - "output": { - "Default/make_tuple[9]_3/make_tuple-op284": { - "shape": [ - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - } - }, - "output_i": 0, - "proxy_input": {}, - "proxy_output": {}, - "subnode_count": 0, - "independent_layout": false, - "watched": 0 - }, - { - "name": "Default/args0", - "type": "Parameter", - "attr": { - "type": "DT_TENSOR[DT_FLOAT32]", - "shape": "[[32, 1, 32, 32]]" - }, - "input": {}, - "output": { - "Default/TransData-op99": { - "shape": [ - [ - 32, - 1, - 32, - 32 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - } - }, - "output_i": 0, - "proxy_input": {}, - "proxy_output": {}, - "subnode_count": 0, - "independent_layout": false, - "watched": 0 - }, - { - "name": "Default/tuple_getitem[10]_0", - "type": "aggregation_scope", - "attr": {}, - "input": { - "Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/ReLUV2-op87": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_UINT8]']" - }, - "Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/MaxPoolWithArgmax-op3": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_UINT16]']" - }, - "Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/ReLUV2-op89": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_UINT8]']" - }, - "Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/MaxPoolWithArgmax-op7": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_UINT16]']" - }, - "Default/network-WithLossCell/_loss_fn-SoftmaxCrossEntropyWithLogits/SoftmaxCrossEntropyWithLogits-op18": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']" - } - }, - "output": { - "Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/Cast-op195": { - "shape": [ - [ - 32, - 6, - 28, - 28 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/network-WithLossCell/_backbone-LeNet5/conv2-Conv2d/Conv2D-op5": { - "shape": [ - [ - 32, - 6, - 14, - 14 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/conv2-Conv2d/gradConv2D/Conv2DBackpropFilter-op48": { - "shape": [ - [ - 32, - 6, - 14, - 14 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/Cast-op205": { - "shape": [ - [ - 32, - 16, - 10, - 10 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/Cast-op208": { - "shape": [ - [ - 32, - 16, - 5, - 5 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_loss_fn-SoftmaxCrossEntropyWithLogits/gradSoftmaxCrossEntropyWithLogits/Mul-op20": { - "shape": [ - [ - 32, - 10 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/gradReLU/ReluGradV2-op92": { - "shape": [ - [ - 32, - 1, - 10, - 10, - 2 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_UINT8]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/gradMaxPoolWithArgmax/MaxPoolGradWithArgmax-op46": { - "shape": [ - [ - 32, - 16, - 4, - 3 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_UINT16]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/gradReLU/ReluGradV2-op94": { - "shape": [ - [ - 32, - 1, - 28, - 28, - 2 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_UINT8]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/gradMaxPoolWithArgmax/MaxPoolGradWithArgmax-op53": { - "shape": [ - [ - 32, - 6, - 4, - 14 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_UINT16]" - }, - "Default/network-WithLossCell/_loss_fn-SoftmaxCrossEntropyWithLogits/ReduceMean-op60": { - "shape": [ - [ - 32 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - } - }, - "output_i": 0, - "proxy_input": {}, - "proxy_output": {}, - "subnode_count": 20, - "independent_layout": false, - "watched": 0 - }, - { - "name": "Default/make_tuple[9]_3", - "type": "aggregation_scope", - "attr": {}, - "input": { - "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op250": { - "shape": [ - [ - 10 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op251": { - "shape": [ - [ - 10 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op253": { - "shape": [ - [ - 10, - 84 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op254": { - "shape": [ - [ - 10, - 84 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op256": { - "shape": [ - [ - 84 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op257": { - "shape": [ - [ - 84 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op259": { - "shape": [ - [ - 84, - 120 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op260": { - "shape": [ - [ - 84, - 120 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op262": { - "shape": [ - [ - 120 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op263": { - "shape": [ - [ - 120 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op265": { - "shape": [ - [ - 120, - 400 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op266": { - "shape": [ - [ - 120, - 400 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op268": { - "shape": [ - [ - 16, - 6, - 5, - 5 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op269": { - "shape": [ - [ - 16, - 6, - 5, - 5 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op271": { - "shape": [ - [ - 6, - 1, - 5, - 5 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op272": { - "shape": [ - [ - 6, - 1, - 5, - 5 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/Depend-op61": { - "shape": [ - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - } - }, - "output": { - "Default/optimizer-Momentum/Depend[8]_4/Depend-op23": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']" - }, - "Default/optimizer-Momentum/Depend[8]_4/Depend-op26": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']" - }, - "Default/optimizer-Momentum/Depend[8]_4/Depend-op31": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']" - }, - "Default/optimizer-Momentum/Depend[8]_4/Depend-op34": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']" - }, - "Default/optimizer-Momentum/Depend[8]_4/Depend-op39": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']" - }, - "Default/optimizer-Momentum/Depend[8]_4/Depend-op42": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']" - }, - "Default/optimizer-Momentum/Depend[8]_4/Depend-op50": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']" - }, - "Default/optimizer-Momentum/Depend[8]_4/Depend-op57": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']" - } - }, - "output_i": 0, - "proxy_input": {}, - "proxy_output": {}, - "subnode_count": 9, - "independent_layout": false, - "watched": 0 - }, - { - "name": "Default/network-WithLossCell", - "type": "name_scope", - "attr": {}, - "input": { - "Default/TransData-op99": { - "shape": [ - [ - 32, - 1, - 32, - 32 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/tuple_getitem[10]_0/tuple_getitem-op193": { - "shape": [ - [ - 32, - 6, - 28, - 28 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/tuple_getitem[10]_0/tuple_getitem-op197": { - "shape": [ - [ - 32, - 6, - 14, - 14 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/tuple_getitem[10]_0/tuple_getitem-op203": { - "shape": [ - [ - 32, - 16, - 10, - 10 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/tuple_getitem[10]_0/tuple_getitem-op207": { - "shape": [ - [ - 32, - 16, - 5, - 5 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/tuple_getitem[10]_0/tuple_getitem-op211": { - "shape": [ - [ - 32 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - } - }, - "output": { - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/conv2-Conv2d/gradConv2D/Conv2DBackpropInput-op52": { - "shape": [ - [ - 16, - 6, - 5, - 5 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/conv1-Conv2d/gradConv2D/Conv2DBackpropFilter-op55": { - "shape": [ - [ - 32, - 1, - 32, - 32 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/tuple_getitem[10]_0/tuple_getitem-op193": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_UINT8]']" - }, - "Default/tuple_getitem[10]_0/tuple_getitem-op192": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_UINT8]']" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/gradMaxPoolWithArgmax/MaxPoolGradWithArgmax-op53": { - "shape": [ - [ - 32, - 6, - 28, - 28 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/tuple_getitem[10]_0/tuple_getitem-op197": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_UINT16]']" - }, - "Default/tuple_getitem[10]_0/tuple_getitem-op196": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_UINT16]']" - }, - "Default/tuple_getitem[10]_0/tuple_getitem-op203": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_UINT8]']" - }, - "Default/tuple_getitem[10]_0/tuple_getitem-op202": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_UINT8]']" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/gradMaxPoolWithArgmax/MaxPoolGradWithArgmax-op46": { - "shape": [ - [ - 32, - 16, - 10, - 10 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/tuple_getitem[10]_0/tuple_getitem-op207": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_UINT16]']" - }, - "Default/tuple_getitem[10]_0/tuple_getitem-op206": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_UINT16]']" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradMatMul/MatMul[6]_5/MatMul-op40": { - "shape": [ - [ - 32, - 400 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/gradReLU/ReluGrad-op36": { - "shape": [ - [ - 32, - 120 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradMatMul/MatMul[6]_5/MatMul-op32": { - "shape": [ - [ - 32, - 120 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/gradReLU/ReluGrad-op28": { - "shape": [ - [ - 32, - 84 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradMatMul/MatMul[6]_5/MatMul-op24": { - "shape": [ - [ - 32, - 84 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/tuple_getitem[10]_0/tuple_getitem-op210": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']" - }, - "Default/tuple_getitem[10]_0/tuple_getitem-op211": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']" - }, - "Default/Depend-op61": { - "shape": [ - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - } - }, - "output_i": 0, - "proxy_input": {}, - "proxy_output": {}, - "subnode_count": 2, - "independent_layout": false, - "watched": 0 - }, - { - "name": "Default/optimizer-Momentum", - "type": "name_scope", - "attr": {}, - "input": { - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradBiasAdd/BiasAddGrad-op21": { - "shape": [ - [ - 10 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradMatMul/MatMul[6]_5/MatMul-op24": { - "shape": [ - [ - 10, - 84 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradBiasAdd/BiasAddGrad-op29": { - "shape": [ - [ - 84 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradMatMul/MatMul[6]_5/MatMul-op32": { - "shape": [ - [ - 84, - 120 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradBiasAdd/BiasAddGrad-op37": { - "shape": [ - [ - 120 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradMatMul/MatMul[6]_5/MatMul-op40": { - "shape": [ - [ - 120, - 400 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/conv2-Conv2d/gradConv2D/Conv2DBackpropFilter-op48": { - "shape": [ - [ - 16, - 6, - 5, - 5 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Gradients/Default/network-WithLossCell/_backbone-LeNet5/conv1-Conv2d/gradConv2D/Conv2DBackpropFilter-op55": { - "shape": [ - [ - 6, - 1, - 5, - 5 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/make_tuple[9]_3/make_tuple-op65": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']" - }, - "Default/make_tuple[9]_3/make_tuple-op68": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']" - }, - "Default/make_tuple[9]_3/make_tuple-op71": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']" - }, - "Default/make_tuple[9]_3/make_tuple-op74": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']" - }, - "Default/make_tuple[9]_3/make_tuple-op77": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']" - }, - "Default/make_tuple[9]_3/make_tuple-op80": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']" - }, - "Default/make_tuple[9]_3/make_tuple-op83": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']" - }, - "Default/make_tuple[9]_3/make_tuple-op86": { - "shape": [ - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']" - } - }, - "output": { - "Default/Depend-op61": { - "shape": [ - [], - [], - [], - [], - [], - [], - [], - [] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TUPLE['DT_BOOL', 'DT_BOOL', 'DT_BOOL', 'DT_BOOL', 'DT_BOOL', 'DT_BOOL', 'DT_BOOL', 'DT_BOOL']" - }, - "Default/make_tuple[9]_3/make_tuple-op65": { - "shape": [ - [ - 10 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/make_tuple[9]_3/make_tuple-op68": { - "shape": [ - [ - 10, - 84 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/make_tuple[9]_3/make_tuple-op71": { - "shape": [ - [ - 84 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/make_tuple[9]_3/make_tuple-op74": { - "shape": [ - [ - 84, - 120 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/make_tuple[9]_3/make_tuple-op77": { - "shape": [ - [ - 120 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/make_tuple[9]_3/make_tuple-op80": { - "shape": [ - [ - 120, - 400 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/make_tuple[9]_3/make_tuple-op83": { - "shape": [ - [ - 16, - 6, - 5, - 5 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - }, - "Default/make_tuple[9]_3/make_tuple-op86": { - "shape": [ - [ - 6, - 1, - 5, - 5 - ] - ], - "edge_type": "data", - "independent_layout": false, - "data_type": "DT_TENSOR[DT_FLOAT32]" - } - }, - "output_i": 0, - "proxy_input": {}, - "proxy_output": {}, - "subnode_count": 5, - "independent_layout": false, - "watched": 1 - } - ], - "scope_name": "Default", - "children": {} - } - } -} \ No newline at end of file +{"tensor_history": [{"name": "Default/TransData-op99:0", "full_name": "Default/TransData-op99:0", "node_type": "TransData", "type": "output", "graph_name": "graph_0", "is_hit": true}, {"name": "Default/args0:0", "full_name": "Default/args0:0", "node_type": "Parameter", "type": "input", "graph_name": "graph_0", "is_hit": false}], "metadata": {"state": "waiting", "step": 2}, "graph": {"nodes": [{"name": "Default", "type": "name_scope", "attr": {}, "input": {"Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradBiasAdd/BiasAddGrad-op21": {"shape": [[10]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradMatMul/MatMul[6]_5/MatMul-op24": {"shape": [[10, 84]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradBiasAdd/BiasAddGrad-op29": {"shape": [[84]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradMatMul/MatMul[6]_5/MatMul-op32": {"shape": [[84, 120]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradBiasAdd/BiasAddGrad-op37": {"shape": [[120]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradMatMul/MatMul[6]_5/MatMul-op40": {"shape": [[120, 400]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/conv2-Conv2d/gradConv2D/Conv2DBackpropFilter-op48": {"shape": [[16, 6, 5, 5]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/conv1-Conv2d/gradConv2D/Conv2DBackpropFilter-op55": {"shape": [[6, 1, 5, 5]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}}, "output": {"Gradients/Default/network-WithLossCell/_backbone-LeNet5/conv2-Conv2d/gradConv2D/Conv2DBackpropInput-op52": {"shape": [[16, 6, 5, 5]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/conv1-Conv2d/gradConv2D/Conv2DBackpropFilter-op55": {"shape": [[32, 1, 32, 32]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/gradMaxPoolWithArgmax/MaxPoolGradWithArgmax-op53": {"shape": [[32, 6, 4, 14]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_UINT16]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/gradMaxPoolWithArgmax/MaxPoolGradWithArgmax-op46": {"shape": [[32, 16, 4, 3]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_UINT16]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradMatMul/MatMul[6]_5/MatMul-op40": {"shape": [[32, 400]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/gradReLU/ReluGrad-op36": {"shape": [[32, 120]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradMatMul/MatMul[6]_5/MatMul-op32": {"shape": [[32, 120]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/gradReLU/ReluGrad-op28": {"shape": [[32, 84]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradMatMul/MatMul[6]_5/MatMul-op24": {"shape": [[32, 84]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/conv2-Conv2d/gradConv2D/Conv2DBackpropFilter-op48": {"shape": [[32, 6, 14, 14]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_loss_fn-SoftmaxCrossEntropyWithLogits/gradSoftmaxCrossEntropyWithLogits/Mul-op20": {"shape": [[32, 10]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/gradReLU/ReluGradV2-op92": {"shape": [[32, 1, 10, 10, 2]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_UINT8]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/gradReLU/ReluGradV2-op94": {"shape": [[32, 1, 28, 28, 2]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_UINT8]"}}, "output_i": 0, "proxy_input": {}, "proxy_output": {}, "subnode_count": 7, "independent_layout": false, "watched": 2}, {"name": "Gradients", "type": "name_scope", "attr": {}, "input": {"Default/tuple_getitem[10]_0/tuple_getitem-op210": {"shape": [[32, 10]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/ReLU-op15": {"shape": [[32, 84]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/ReLU-op12": {"shape": [[32, 120]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/Cast-op205": {"shape": [[32, 16, 10, 10]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/tuple_getitem[10]_0/tuple_getitem-op206": {"shape": [[32, 16, 4, 3]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_UINT16]"}, "Default/tuple_getitem[10]_0/tuple_getitem-op202": {"shape": [[32, 1, 10, 10, 2]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_UINT8]"}, "Default/tuple_getitem[10]_0/tuple_getitem-op197": {"shape": [[32, 6, 14, 14]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/network-WithLossCell/_backbone-LeNet5/conv2-Conv2d/Cast-op188": {"shape": [[16, 6, 5, 5]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/Cast-op195": {"shape": [[32, 6, 28, 28]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/tuple_getitem[10]_0/tuple_getitem-op196": {"shape": [[32, 6, 4, 14]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_UINT16]"}, "Default/tuple_getitem[10]_0/tuple_getitem-op192": {"shape": [[32, 1, 28, 28, 2]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_UINT8]"}, "Default/network-WithLossCell/_backbone-LeNet5/conv1-Conv2d/Cast-op190": {"shape": [[32, 1, 32, 32]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/network-WithLossCell/_backbone-LeNet5/flatten-Flatten/Reshape-op9": {"shape": [[32, 400]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}}, "output": {"Default/optimizer-Momentum/ApplyMomentum[8]_1/ApplyMomentum-op22": {"shape": [[10]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/optimizer-Momentum/ApplyMomentum[8]_1/ApplyMomentum-op30": {"shape": [[84]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/optimizer-Momentum/ApplyMomentum[8]_1/ApplyMomentum-op38": {"shape": [[120]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/optimizer-Momentum/ApplyMomentum[8]_1/ApplyMomentum-op49": {"shape": [[16, 6, 5, 5]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/optimizer-Momentum/ApplyMomentum[8]_1/ApplyMomentum-op56": {"shape": [[6, 1, 5, 5]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/optimizer-Momentum/ApplyMomentum[8]_1/ApplyMomentum-op25": {"shape": [[10, 84]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/optimizer-Momentum/ApplyMomentum[8]_1/ApplyMomentum-op33": {"shape": [[84, 120]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/optimizer-Momentum/ApplyMomentum[8]_1/ApplyMomentum-op41": {"shape": [[120, 400]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}}, "output_i": 0, "proxy_input": {}, "proxy_output": {}, "subnode_count": 1, "independent_layout": false, "watched": 2}], "scope_name": "", "children": {"nodes": [{"name": "Default/TransData-op99", "type": "TransData", "attr": {"dst_format": "dtype: DT_STRING\nstr_val: \"NC1HWC0\"\n", "IsFeatureMapInputList": "dtype: DT_TUPLE\nvalues {\n dtype: DT_UINT64\n uint_val: 0\n}\n", "pri_format": "dtype: DT_STRING\nstr_val: \"NC1HWC0\"\n", "_datadump_original_names": "dtype: DT_TUPLE\n", "IsFeatureMapOutput": "dtype: DT_BOOL\nbool_val: true\n", "src_format": "dtype: DT_STRING\nstr_val: \"NCHW\"\n"}, "input": {"Default/args0": {"shape": [[32, 1, 32, 32]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}}, "output": {"Default/network-WithLossCell/_backbone-LeNet5/conv1-Conv2d/Cast-op190": {"shape": [[32, 1, 32, 32]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}}, "output_i": 0, "proxy_input": {}, "proxy_output": {}, "subnode_count": 0, "independent_layout": false, "watched": 2}, {"name": "Default/Depend-op61", "type": "Depend", "attr": {}, "input": {"Default/network-WithLossCell/_loss_fn-SoftmaxCrossEntropyWithLogits/ReduceMean-op60": {"shape": [[]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/optimizer-Momentum/make_tuple-op58": {"shape": [[], [], [], [], [], [], [], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_BOOL', 'DT_BOOL', 'DT_BOOL', 'DT_BOOL', 'DT_BOOL', 'DT_BOOL', 'DT_BOOL', 'DT_BOOL']"}}, "output": {"Default/make_tuple[9]_3/make_tuple-op284": {"shape": [[]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}}, "output_i": 0, "proxy_input": {}, "proxy_output": {}, "subnode_count": 0, "independent_layout": false, "watched": 2}, {"name": "Default/args0", "type": "Parameter", "attr": {"type": "DT_TENSOR[DT_FLOAT32]", "shape": "[[32, 1, 32, 32]]"}, "input": {}, "output": {"Default/TransData-op99": {"shape": [[32, 1, 32, 32]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}}, "output_i": 0, "proxy_input": {}, "proxy_output": {}, "subnode_count": 0, "independent_layout": false, "watched": 2}, {"name": "Default/tuple_getitem[10]_0", "type": "aggregation_scope", "attr": {}, "input": {"Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/ReLUV2-op87": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_UINT8]']"}, "Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/MaxPoolWithArgmax-op3": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_UINT16]']"}, "Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/ReLUV2-op89": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_UINT8]']"}, "Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/MaxPoolWithArgmax-op7": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_UINT16]']"}, "Default/network-WithLossCell/_loss_fn-SoftmaxCrossEntropyWithLogits/SoftmaxCrossEntropyWithLogits-op18": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']"}}, "output": {"Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/Cast-op195": {"shape": [[32, 6, 28, 28]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/network-WithLossCell/_backbone-LeNet5/conv2-Conv2d/Conv2D-op5": {"shape": [[32, 6, 14, 14]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/conv2-Conv2d/gradConv2D/Conv2DBackpropFilter-op48": {"shape": [[32, 6, 14, 14]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/Cast-op205": {"shape": [[32, 16, 10, 10]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/Cast-op208": {"shape": [[32, 16, 5, 5]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_loss_fn-SoftmaxCrossEntropyWithLogits/gradSoftmaxCrossEntropyWithLogits/Mul-op20": {"shape": [[32, 10]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/gradReLU/ReluGradV2-op92": {"shape": [[32, 1, 10, 10, 2]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_UINT8]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/gradMaxPoolWithArgmax/MaxPoolGradWithArgmax-op46": {"shape": [[32, 16, 4, 3]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_UINT16]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/gradReLU/ReluGradV2-op94": {"shape": [[32, 1, 28, 28, 2]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_UINT8]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/gradMaxPoolWithArgmax/MaxPoolGradWithArgmax-op53": {"shape": [[32, 6, 4, 14]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_UINT16]"}, "Default/network-WithLossCell/_loss_fn-SoftmaxCrossEntropyWithLogits/ReduceMean-op60": {"shape": [[32]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}}, "output_i": 0, "proxy_input": {}, "proxy_output": {}, "subnode_count": 20, "independent_layout": false, "watched": 2}, {"name": "Default/make_tuple[9]_3", "type": "aggregation_scope", "attr": {}, "input": {"Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op250": {"shape": [[10]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op251": {"shape": [[10]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op253": {"shape": [[10, 84]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op254": {"shape": [[10, 84]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op256": {"shape": [[84]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op257": {"shape": [[84]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op259": {"shape": [[84, 120]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op260": {"shape": [[84, 120]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op262": {"shape": [[120]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op263": {"shape": [[120]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op265": {"shape": [[120, 400]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op266": {"shape": [[120, 400]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op268": {"shape": [[16, 6, 5, 5]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op269": {"shape": [[16, 6, 5, 5]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op271": {"shape": [[6, 1, 5, 5]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/optimizer-Momentum/tuple_getitem[16]_2/tuple_getitem-op272": {"shape": [[6, 1, 5, 5]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/Depend-op61": {"shape": [[]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}}, "output": {"Default/optimizer-Momentum/Depend[8]_4/Depend-op23": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']"}, "Default/optimizer-Momentum/Depend[8]_4/Depend-op26": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']"}, "Default/optimizer-Momentum/Depend[8]_4/Depend-op31": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']"}, "Default/optimizer-Momentum/Depend[8]_4/Depend-op34": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']"}, "Default/optimizer-Momentum/Depend[8]_4/Depend-op39": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']"}, "Default/optimizer-Momentum/Depend[8]_4/Depend-op42": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']"}, "Default/optimizer-Momentum/Depend[8]_4/Depend-op50": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']"}, "Default/optimizer-Momentum/Depend[8]_4/Depend-op57": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']"}}, "output_i": 0, "proxy_input": {}, "proxy_output": {}, "subnode_count": 9, "independent_layout": false, "watched": 2}, {"name": "Default/network-WithLossCell", "type": "name_scope", "attr": {}, "input": {"Default/TransData-op99": {"shape": [[32, 1, 32, 32]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/tuple_getitem[10]_0/tuple_getitem-op193": {"shape": [[32, 6, 28, 28]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/tuple_getitem[10]_0/tuple_getitem-op197": {"shape": [[32, 6, 14, 14]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/tuple_getitem[10]_0/tuple_getitem-op203": {"shape": [[32, 16, 10, 10]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/tuple_getitem[10]_0/tuple_getitem-op207": {"shape": [[32, 16, 5, 5]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/tuple_getitem[10]_0/tuple_getitem-op211": {"shape": [[32]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}}, "output": {"Gradients/Default/network-WithLossCell/_backbone-LeNet5/conv2-Conv2d/gradConv2D/Conv2DBackpropInput-op52": {"shape": [[16, 6, 5, 5]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/conv1-Conv2d/gradConv2D/Conv2DBackpropFilter-op55": {"shape": [[32, 1, 32, 32]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/tuple_getitem[10]_0/tuple_getitem-op193": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_UINT8]']"}, "Default/tuple_getitem[10]_0/tuple_getitem-op192": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_UINT8]']"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/gradMaxPoolWithArgmax/MaxPoolGradWithArgmax-op53": {"shape": [[32, 6, 28, 28]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/tuple_getitem[10]_0/tuple_getitem-op197": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_UINT16]']"}, "Default/tuple_getitem[10]_0/tuple_getitem-op196": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_UINT16]']"}, "Default/tuple_getitem[10]_0/tuple_getitem-op203": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_UINT8]']"}, "Default/tuple_getitem[10]_0/tuple_getitem-op202": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_UINT8]']"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/gradMaxPoolWithArgmax/MaxPoolGradWithArgmax-op46": {"shape": [[32, 16, 10, 10]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/tuple_getitem[10]_0/tuple_getitem-op207": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_UINT16]']"}, "Default/tuple_getitem[10]_0/tuple_getitem-op206": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_UINT16]']"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradMatMul/MatMul[6]_5/MatMul-op40": {"shape": [[32, 400]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/gradReLU/ReluGrad-op36": {"shape": [[32, 120]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradMatMul/MatMul[6]_5/MatMul-op32": {"shape": [[32, 120]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/gradReLU/ReluGrad-op28": {"shape": [[32, 84]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradMatMul/MatMul[6]_5/MatMul-op24": {"shape": [[32, 84]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/tuple_getitem[10]_0/tuple_getitem-op210": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']"}, "Default/tuple_getitem[10]_0/tuple_getitem-op211": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']"}, "Default/Depend-op61": {"shape": [[]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}}, "output_i": 0, "proxy_input": {}, "proxy_output": {}, "subnode_count": 2, "independent_layout": false, "watched": 2}, {"name": "Default/optimizer-Momentum", "type": "name_scope", "attr": {}, "input": {"Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradBiasAdd/BiasAddGrad-op21": {"shape": [[10]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradMatMul/MatMul[6]_5/MatMul-op24": {"shape": [[10, 84]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradBiasAdd/BiasAddGrad-op29": {"shape": [[84]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradMatMul/MatMul[6]_5/MatMul-op32": {"shape": [[84, 120]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradBiasAdd/BiasAddGrad-op37": {"shape": [[120]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/gradMatMul/MatMul[6]_5/MatMul-op40": {"shape": [[120, 400]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/conv2-Conv2d/gradConv2D/Conv2DBackpropFilter-op48": {"shape": [[16, 6, 5, 5]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Gradients/Default/network-WithLossCell/_backbone-LeNet5/conv1-Conv2d/gradConv2D/Conv2DBackpropFilter-op55": {"shape": [[6, 1, 5, 5]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/make_tuple[9]_3/make_tuple-op65": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']"}, "Default/make_tuple[9]_3/make_tuple-op68": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']"}, "Default/make_tuple[9]_3/make_tuple-op71": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']"}, "Default/make_tuple[9]_3/make_tuple-op74": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']"}, "Default/make_tuple[9]_3/make_tuple-op77": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']"}, "Default/make_tuple[9]_3/make_tuple-op80": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']"}, "Default/make_tuple[9]_3/make_tuple-op83": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']"}, "Default/make_tuple[9]_3/make_tuple-op86": {"shape": [[], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_TENSOR[DT_FLOAT32]', 'DT_TENSOR[DT_FLOAT32]']"}}, "output": {"Default/Depend-op61": {"shape": [[], [], [], [], [], [], [], []], "edge_type": "data", "independent_layout": false, "data_type": "DT_TUPLE['DT_BOOL', 'DT_BOOL', 'DT_BOOL', 'DT_BOOL', 'DT_BOOL', 'DT_BOOL', 'DT_BOOL', 'DT_BOOL']"}, "Default/make_tuple[9]_3/make_tuple-op65": {"shape": [[10]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/make_tuple[9]_3/make_tuple-op68": {"shape": [[10, 84]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/make_tuple[9]_3/make_tuple-op71": {"shape": [[84]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/make_tuple[9]_3/make_tuple-op74": {"shape": [[84, 120]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/make_tuple[9]_3/make_tuple-op77": {"shape": [[120]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/make_tuple[9]_3/make_tuple-op80": {"shape": [[120, 400]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/make_tuple[9]_3/make_tuple-op83": {"shape": [[16, 6, 5, 5]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}, "Default/make_tuple[9]_3/make_tuple-op86": {"shape": [[6, 1, 5, 5]], "edge_type": "data", "independent_layout": false, "data_type": "DT_TENSOR[DT_FLOAT32]"}}, "output_i": 0, "proxy_input": {}, "proxy_output": {}, "subnode_count": 5, "independent_layout": false, "watched": 2}], "scope_name": "Default", "children": {}}}} \ No newline at end of file diff --git a/tests/st/func/debugger/expect_results/restful_results/search_unwatched_leaf_node.json b/tests/st/func/debugger/expect_results/restful_results/search_unwatched_leaf_node.json index f25fae41..12d7222b 100644 --- a/tests/st/func/debugger/expect_results/restful_results/search_unwatched_leaf_node.json +++ b/tests/st/func/debugger/expect_results/restful_results/search_unwatched_leaf_node.json @@ -1,31 +1 @@ -{ - "nodes": [ - { - "name": "Default", - "type": "name_scope", - "nodes": [ - { - "name": "Default/optimizer-Momentum", - "type": "name_scope", - "nodes": [ - { - "name": "Default/optimizer-Momentum/Parameter[18]_7", - "type": "aggregation_scope", - "nodes": [ - { - "name": "Default/optimizer-Momentum/Parameter[18]_7/moments.fc3.bias", - "type": "Parameter", - "nodes": [], - "watched": 0 - } - ], - "watched": 0 - } - ], - "watched": 0 - } - ], - "watched": 0 - } - ] -} \ No newline at end of file +{"nodes": [{"name": "Default", "type": "name_scope", "nodes": [{"name": "Default/optimizer-Momentum", "type": "name_scope", "nodes": [{"name": "Default/optimizer-Momentum/Parameter[18]_7", "type": "aggregation_scope", "nodes": [{"name": "Default/optimizer-Momentum/Parameter[18]_7/moments.fc3.bias", "type": "Parameter", "nodes": [], "watched": 2}], "watched": 2}], "watched": 2}], "watched": 2}]} \ No newline at end of file diff --git a/tests/st/func/debugger/test_restful_api.py b/tests/st/func/debugger/test_restful_api.py index 087b77c8..f0b69b56 100644 --- a/tests/st/func/debugger/test_restful_api.py +++ b/tests/st/func/debugger/test_restful_api.py @@ -169,7 +169,7 @@ class TestAscendDebugger: url = 'update-watchpoint' body_data = {'watch_point_id': watch_point_id, 'watch_nodes': [leaf_node_name], - 'mode': 0} + 'mode': 1} get_request_result(app_client, url, body_data) # get updated nodes url = 'search' @@ -328,7 +328,7 @@ class TestAscendDebugger: 'watch_nodes': ['Default']}, True), ('update-watchpoint', {'watch_point_id': 1, 'watch_nodes': ['Default/optimizer-Momentum/Parameter[18]_7'], - 'mode': 0}, True), + 'mode': 1}, True), ('update-watchpoint', {'watch_point_id': 1, 'watch_nodes': ['Default/optimizer-Momentum'], 'mode': 1}, True), @@ -438,7 +438,7 @@ class TestGPUDebugger: 'watch_nodes': ['Default/TransData-op99']}, True), ('update-watchpoint', {'watch_point_id': 1, 'watch_nodes': ['Default/optimizer-Momentum/Parameter[18]_7'], - 'mode': 0}, True), + 'mode': 1}, True), ('update-watchpoint', {'watch_point_id': 1, 'watch_nodes': ['Default/optimizer-Momentum'], 'mode': 1}, True), @@ -450,9 +450,9 @@ class TestGPUDebugger: ], True), ('update-watchpoint', [{'watch_point_id': 1, 'watch_nodes': ['Default/TransData-op99'], - 'mode': 0}, + 'mode': 1}, {'watch_point_id': 1, 'watch_nodes': ['Default/TransData-op99'], - 'mode': 1} + 'mode': 0} ], True), ('delete-watchpoint', {'watch_point_id': 1}, True) ]) diff --git a/tests/ut/debugger/test_debugger_server.py b/tests/ut/debugger/test_debugger_server.py index 4e9e67c3..506eeeb8 100644 --- a/tests/ut/debugger/test_debugger_server.py +++ b/tests/ut/debugger/test_debugger_server.py @@ -32,6 +32,7 @@ from mindinsight.debugger.common.utils import Streams from mindinsight.debugger.debugger_cache import DebuggerCache from mindinsight.debugger.debugger_server import DebuggerServer from mindinsight.debugger.debugger_server import grpc_server_base +from mindinsight.debugger.stream_operator import watchpoint_operator from mindinsight.debugger.stream_handler import GraphHandler, WatchpointHandler, MetadataHandler, \ TensorHandler from tests.ut.debugger.configurations import compare_debugger_result_with_file, mock_tensor_history @@ -196,6 +197,7 @@ class TestDebuggerServer: @mock.patch.object(MetadataHandler, 'backend', 'GPU') @mock.patch.object(GraphHandler, 'get_node_basic_info', return_value=MagicMock()) @mock.patch.object(GraphHandler, 'get_node_type', return_value='aggregation_scope') + @mock.patch.object(watchpoint_operator, 'get_basic_node_info', return_value=MagicMock()) @mock.patch.object(WatchpointHandler, 'create_watchpoint') def test_create_watchpoint(self, *args): """Test create watchpoint."""