Searched refs:_analyze_einsum_string (Results 1 – 2 of 2) sorted by relevance
142 shape_data = _analyze_einsum_string(self.equation,209 def _analyze_einsum_string(equation, bias_axes, input_shape, output_shape): function
232 weight_shape, bias_shape, _ = einsum_dense._analyze_einsum_string(