Searched refs:check_op_name (Results 1 – 3 of 3) sorted by relevance
387 is_training_mode_flag=timeline_generator.check_op_name('Gradients'),438 skip_first_step_flag = framework_parser.check_op_name(INIT_OP_NAME)441 is_training_mode_flag = framework_parser.check_op_name("Gradients")
244 def check_op_name(self, op_name, is_prefix=True): member in FrameworkParser
983 def check_op_name(self, op_name): member in GpuTimelineGenerator