Searched refs:dump_option_ (Results 1 – 2 of 2) sorted by relevance
144 …return dump_option_.is_before_select_kernel ? AnfAlgo::GetPrevNodeOutputInferDataType(anf_node, re… in GetInputDataType()149 …return dump_option_.is_before_select_kernel ? AnfAlgo::GetPrevNodeOutputInferShape(anf_node, real_… in GetInputShape()154 …return dump_option_.is_before_select_kernel ? kOpFormat_DEFAULT : AnfAlgo::GetInputFormat(anf_node… in GetInputFormat()158 return dump_option_.is_before_select_kernel ? AnfAlgo::GetOutputInferDataType(anf_node, index) in GetOutputDataType()163 return dump_option_.is_before_select_kernel ? AnfAlgo::GetOutputInferShape(anf_node, index) in GetOutputShape()168 …return dump_option_.is_before_select_kernel ? kOpFormat_DEFAULT : AnfAlgo::GetOutputFormat(anf_nod… in GetOutputFormat()194 std::string dtype = TypeId2String(type_id, dump_option_.is_before_select_kernel); in CreateInputDescJson()235 std::string dtype = TypeId2String(type_id, dump_option_.is_before_select_kernel); in CreateOutputDescJson()441 if (dump_option_.save_ptr_address) { in SaveNodeAddress()451 if (dump_option_.extract_opinfo_from_anfnode) { in ExtractOpInfo()[all …]
92 explicit AkgKernelJsonGenerator(DumpOption dump_option) : dump_option_(dump_option) { Clear(); } in AkgKernelJsonGenerator()112 void set_dump_option(DumpOption dump_option) { dump_option_ = dump_option; } in set_dump_option()153 DumpOption dump_option_; variable