Searched refs:hint_type (Results 1 – 7 of 7) sorted by relevance
28 def cast_to_tensor(t, hint_type=mstype.float32): argument50 return Tensor(t, dtype=hint_type)333 def set_param_type(args, hint_type): argument345 if hint_type in int_type or hint_type is None:346 hint_type = mstype.float32360 return hint_type if common_dtype is None else common_dtype
823 PS_Hint_Type hint_type ) in ps_hints_open() argument826 hints->hint_type = hint_type; in ps_hints_open()909 if ( hints->hint_type == PS_HINT_TYPE_1 ) in ps_hints_t1stem3()957 if ( hints->hint_type == PS_HINT_TYPE_1 ) in ps_hints_t1reset()
131 PS_Hint_Type hint_type; member
828 PS_Hint_Type hint_type ) in ps_hints_open() argument831 hints->hint_type = hint_type; in ps_hints_open()914 if ( hints->hint_type == PS_HINT_TYPE_1 ) in ps_hints_t1stem3()962 if ( hints->hint_type == PS_HINT_TYPE_1 ) in ps_hints_t1reset()
829 PS_Hint_Type hint_type ) in ps_hints_open() argument832 hints->hint_type = hint_type; in ps_hints_open()915 if ( hints->hint_type == PS_HINT_TYPE_1 ) in ps_hints_t1stem3()963 if ( hints->hint_type == PS_HINT_TYPE_1 ) in ps_hints_t1reset()
132 PS_Hint_Type hint_type; member