Searched refs:dim_axis (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/reduce_slice_ops/ops/ |
| D | reduce_slice_ops.cc | 30 DimensionHandle dim_axis = c->UnknownDim(); in ReduceSliceShapeFn() local
45 TF_RETURN_IF_ERROR(c->Subtract(raw_dim_axis, 1, &dim_axis)); in ReduceSliceShapeFn()
49 dim_axis = c->Dim(c->input(1), 0); in ReduceSliceShapeFn()
58 TF_RETURN_IF_ERROR(c->ReplaceDim(handle, axis, dim_axis, &handle)); in ReduceSliceShapeFn()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 2858 dim_axis = dim % shape.ndims
2859 logits = _swap_axis(logits, dim_axis, math_ops.subtract(input_rank, 1))
2865 output, dim_axis, math_ops.subtract(input_rank, 1), name=name)
|