Searched refs:expanded_num_dims (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | pack_op.cc | 63 int expanded_num_dims = shapes[0].dims() + 1; in Compile() local
65 if (axis < 0) axis += expanded_num_dims; in Compile()
67 OP_REQUIRES(ctx, 0 <= axis && axis < expanded_num_dims, in Compile()
69 -expanded_num_dims, ", ", in Compile()
70 expanded_num_dims, ")")); in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pack_op.cc | 53 int expanded_num_dims = first_input.dims() + 1; in Compute() local
55 if (axis < 0) axis += expanded_num_dims; in Compute()
57 OP_REQUIRES(c, 0 <= axis && axis < expanded_num_dims, in Compute()
59 -expanded_num_dims, ", ", in Compute()
60 expanded_num_dims, ")")); in Compute()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 1418 expanded_num_dims = len(value_shape) + 1
1419 if axis < -expanded_num_dims or axis >= expanded_num_dims:
1421 (axis, -expanded_num_dims, expanded_num_dims))
|