Searched refs:ConstantInputAsPartialShape (Results 1 – 3 of 3) sorted by relevance
158 Status ConstantInputAsPartialShape(int index, PartialTensorShape* shape);
414 Status XlaOpKernelContext::ConstantInputAsPartialShape( in ConstantInputAsPartialShape() function in tensorflow::XlaOpKernelContext
123 TF_RETURN_IF_ERROR(ctx->ConstantInputAsPartialShape(0, &partial_shape)); in TryGetElementShapeFromInput()