HomeSort by relevance Sort by last modified time
    Searched refs:expanded_num_dims (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
pack_op.cc 63 int expanded_num_dims = shapes[0].dims() + 1; variable
65 if (axis < 0) axis += expanded_num_dims;
67 OP_REQUIRES(ctx, 0 <= axis && axis < expanded_num_dims,
69 -expanded_num_dims, ", ",
70 expanded_num_dims, ")"));
  /external/tensorflow/tensorflow/core/kernels/
pack_op.cc 66 int expanded_num_dims = values[0].dims() + 1; variable
68 if (axis < 0) axis += expanded_num_dims;
70 OP_REQUIRES(c, 0 <= axis && axis < expanded_num_dims,
72 -expanded_num_dims, ", ",
73 expanded_num_dims, ")"));
  /external/tensorflow/tensorflow/python/ops/
array_ops.py 935 expanded_num_dims = value_shape.ndims + 1
936 if axis < -expanded_num_dims or axis >= expanded_num_dims:
937 raise ValueError("axis = %d not in [%d, %d)" % (axis, -expanded_num_dims,
938 expanded_num_dims))
    [all...]

Completed in 68 milliseconds