Home
last modified time | relevance | path

Searched refs:ndims_name (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/ops/structured/
Dstructured_array_ops.py123 axis, st.rank + 1, axis_name='axis', ndims_name='rank(st)')
Dstructured_tensor.py1010 ndims_name='rank(self)')
1015 ndims_name='rank(self)')
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_gather_ops.py96 ndims_name='rank(indices)')
102 axis, params.shape.rank, ndims_name='rank(params)')
Dragged_math_ops.py536 axis, rt_input.shape.ndims, ndims_name='rank(input_tensor)')
Dragged_tensor.py1211 axis, self.shape.rank, ndims_name="rank(self)")
1425 ndims_name="rank(self)")
1430 ndims_name="rank(self)")
Dragged_array_ops.py441 axis = array_ops.get_positive_axis(axis, ndims, ndims_name='rank(input)')
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py6336 def get_positive_axis(axis, ndims, axis_name="axis", ndims_name="ndims"): argument
6371 (axis_name, ndims_name))
6429 axis = get_positive_axis(axis, data.shape.rank, ndims_name="rank(data)")