Searched refs:ConstantInputAsPartialShape (Results 1 – 3 of 3) sorted by relevance
74 TF_RETURN_IF_ERROR(ctx->ConstantInputAsPartialShape( in CreateZerosList()274 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsPartialShape(1, &element_shape)); in Compile()
143 Status ConstantInputAsPartialShape(int index, PartialTensorShape* shape);
322 Status XlaOpKernelContext::ConstantInputAsPartialShape( in ConstantInputAsPartialShape() function in tensorflow::XlaOpKernelContext