Searched refs:define_custom_tensor_wrapper (Results 1 – 2 of 2) sorted by relevance
135 unsqueeze_output_tensor_wrapper = self.define_custom_tensor_wrapper(219 conv_output_tensor_wrapper = self.define_custom_tensor_wrapper(
250 def define_custom_tensor_wrapper( member in NodeVisitor