dace.frontend.tensorflow package¶
Submodules¶
dace.frontend.tensorflow.tensorflow module¶
dace.frontend.tensorflow.winograd module¶
-
dace.frontend.tensorflow.winograd.add_cublas_cusolver(sdfg: dace.sdfg.sdfg.SDFG)¶ Add CUBLAS and CUSOLVER handles to the SDFG.
-
dace.frontend.tensorflow.winograd.mm(state, A_node, B_node, C_node, A_mode: str = 'N', B_mode: str = 'N', label: str = None, A_subset=None, B_subset=None, C_subset=None, A_memlet=None, B_memlet=None, C_memlet=None, map_entry=None, map_exit=None, shadow_a=False, shadow_b=False, buffer_a=False, buffer_c=False)¶
-
dace.frontend.tensorflow.winograd.mm_small(state, A_node, B_node, C_node, A_subset=None, B_subset=None, C_subset=None, A_memlet=None, B_memlet=None, C_memlet=None, map_entry=None, map_exit=None, A_direct=True, B_direct=True)¶
-
dace.frontend.tensorflow.winograd.printer(*inp)¶
-
dace.frontend.tensorflow.winograd.string_builder(string)¶ To match DaCe variable naming conventions, replaces all undesired characters with “_”.
-
dace.frontend.tensorflow.winograd.winograd_convolution(dace_session, tf_node)¶