Searched refs:sizes_shape (Results 1 – 3 of 3) sorted by relevance
38 const TensorShape sizes_shape = ctx->InputShape(1); in Compile() local40 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(sizes_shape), in Compile()42 sizes_shape.DebugString())); in Compile()43 const int64 num_dims = sizes_shape.num_elements(); in Compile()
734 TensorShape sizes_shape = shape_prefix; in Compute() local735 sizes_shape.AddDim(field_count); in Compute()737 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, sizes_shape, &sizes_tensor)); in Compute()
1422 ShapeHandle sizes_shape; in SliceShape() local1423 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &sizes_shape)); in SliceShape()1426 TF_RETURN_IF_ERROR(c->Merge(begin_shape, sizes_shape, &begin_shape)); in SliceShape()