Home
last modified time | relevance | path

Searched refs:axes_dense (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dreverse_op.cc250 gtl::InlinedVector<bool, 8> axes_dense(input_dims, false); in Compute() local
258 OP_REQUIRES(context, !axes_dense[canonical_axis], in Compute()
261 axes_dense[canonical_axis] = true; in Compute()
279 HandleReverseV2Case<Device, T, NDIMS>(context, axes_dense, output); \ in Compute()
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc945 std::vector<bool> axes_dense(c->Rank(input), false); in __anon7c94107b1402() local
954 if (axes_dense[canonical_axis]) { in __anon7c94107b1402()
958 axes_dense[canonical_axis] = true; in __anon7c94107b1402()