Searched refs:dimension_shape (Results 1 – 3 of 3) sorted by relevance
38 const TensorShape dimension_shape = ctx->InputShape(1); in Compile() local40 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(dimension_shape), in Compile()43 dimension_shape.DebugString())); in Compile()
44 const TensorShape dimension_shape = ctx->InputShape(1); in Compile() local45 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(dimension_shape), in Compile()48 dimension_shape.DebugString())); in Compile()
909 ShapeHandle dimension_shape; in ArgOpShape() local910 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &dimension_shape)); in ArgOpShape()