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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
reverse_op.cc 78 const TensorShape axes_shape = ctx->InputShape(1); variable
80 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(axes_shape),
82 axes_shape.DebugString()));
83 OP_REQUIRES(ctx, axes_shape.num_elements() <= x_shape.dims(),
84 errors::InvalidArgument("axes ", axes_shape.DebugString(),
89 if (axes_shape.num_elements() == 0) {
  /external/tensorflow/tensorflow/python/ops/
math_ops.py     [all...]

Completed in 1506 milliseconds