Searched refs:GetElementUnexhaustive (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 159 bool GetElementUnexhaustive(const Tensor& t, int i, const std::set<int>& dtypes, in GetElementUnexhaustive() function
2487 if (!GetElementUnexhaustive(pow, i, {pow_props.dtype()}, &curr)) { in TrySimplify()
2654 if (!GetElementUnexhaustive(constant, k, in TrySimplifyInternal()
2740 if (!GetElementUnexhaustive(constant, k, in TrySimplify()
|