Searched refs:tensor_rank (Results 1 – 25 of 38) sorted by relevance
12
145 if scale.tensor_rank is not None:146 batch_ndims = scale.tensor_rank - 2
241 if scale.tensor_rank is not None:242 batch_ndims = scale.tensor_rank - 2
493 if operator.tensor_rank is None or operator.tensor_rank % 2:569 if operator.tensor_rank is None or operator.tensor_rank % 2:
337 def tensor_rank(self, name="tensor_rank"): member in LinearOperator368 if self.tensor_rank is not None:369 return ops.convert_to_tensor(self.tensor_rank)
1161 tensor_rank = array_ops.rank(tensor)1162 reduce_axis = math_ops.range(2, tensor_rank)1164 tensor_rank > 2,
56 name: "tensor_rank"
59 name: "tensor_rank"
55 name: "tensor_rank"
60 name: "tensor_rank"
54 name: "tensor_rank"