Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_squeeze_op.py65 dense_dims = []
74 dense_dims.append(dim - input.ragged_rank)
100 if dense_dims:
102 squeezed_rt = array_ops.squeeze(squeezed_rt, dense_dims)
/external/tensorflow/tensorflow/core/ops/
Dragged_conversion_ops.cc175 DimensionHandle dense_dims = in RaggedTensorToSparseShapeFn() local
181 c->set_output(0, c->Matrix(num_values, dense_dims)); // indices in RaggedTensorToSparseShapeFn()
183 c->set_output(2, c->Vector(dense_dims)); // dense_shape in RaggedTensorToSparseShapeFn()