Searched refs:reduced_dims (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | linalg_grad.py | 140 reduced_dims = array_ops.stack([input_shape[ax] for ax in reduced_axes])
141 return reduced_subs, reduced_dims, reduced_axes
158 reduced_subs, reduced_dims, reduced_axes = _GetReducedSubscripts(
198 [reduced_dims, array_ops.shape(output_grad)], axis=0)
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorReduction.h | 60 OutputDims* output_dims, ReducedDims* reduced_dims) {
66 (*reduced_dims)[reduceIndex] = input_dims[i];
79 Sizes<>*, array<Index, Rank>* reduced_dims) {
82 (*reduced_dims)[i] = input_dims[i];
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 1332 absl::flat_hash_set<int64> reduced_dims(reduce.dimensions().begin(), in ReductionPreservesLayout() local
1340 if (reduced_dims.contains(i)) { in ReductionPreservesLayout()
1353 if (!reduced_dims.contains(operand_dim)) { in ReductionPreservesLayout()
|