Searched refs:SetDumpShape (Results 1 – 1 of 1) sorted by relevance
395 void SetDumpShape(const std::vector<size_t> &ms_shape, NotNull<aicpu::dump::Shape *> dump_shape) { in SetDumpShape() function424 SetDumpShape(output_shape, NOT_NULL(output.mutable_shape())); in DumpKernelOutput()425 SetDumpShape(output_origin_shape, NOT_NULL(output.mutable_origin_shape())); in DumpKernelOutput()469 SetDumpShape(output_shape, NOT_NULL(input.mutable_shape())); in DumpKernelInput()470 SetDumpShape(output_origin_shape, NOT_NULL(input.mutable_origin_shape())); in DumpKernelInput()