Searched refs:index_is_reduced (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reduce_join_op.cc | 91 void MakeUnreducedIndices(gtl::InlinedVector<bool, 8> index_is_reduced, in MakeUnreducedIndices() argument
95 if (!index_is_reduced[index]) unreduced_indices->push_back(index); in MakeUnreducedIndices()
99 TensorShape GetOutputShape(gtl::InlinedVector<bool, 8> index_is_reduced, in GetOutputShape() argument
102 for (size_t index = 0; index < index_is_reduced.size(); ++index) { in GetOutputShape()
103 if (index_is_reduced[index]) { in GetOutputShape()
133 gtl::InlinedVector<bool, 8> index_is_reduced(input_dims, false); in Compute() local
142 OP_REQUIRES(context, !index_is_reduced[true_reduce_index], in Compute()
145 index_is_reduced[true_reduce_index] = true; in Compute()
151 MakeUnreducedIndices(index_is_reduced, input_dims, &unreduced_indices); in Compute()
156 GetOutputShape(index_is_reduced, input_shape, keep_dims_); in Compute()
|