Searched refs:AsXlaOp (Results 1 – 6 of 6) sorted by relevance
78 TEST_F(XlaExpressionTest, AsXlaOp) { in TEST_F() argument80 XlaExpression::XlaOp(op_, DT_INT32).AsXlaOp(builder_.get()); in TEST_F()84 XlaExpression::Constant(constant_).AsXlaOp(builder_.get()); in TEST_F()
111 xla::XlaOp AsXlaOp(xla::XlaBuilder* builder) const;
62 return InputExpression(index).AsXlaOp(builder()); in Input()66 return InputExpression(name).AsXlaOp(builder()); in Input()444 XlaExpression::CastExpressionFromTensor(input)->AsXlaOp(builder())); in InputList()
90 xla::XlaOp XlaExpression::AsXlaOp(xla::XlaBuilder* builder) const { in AsXlaOp() function in tensorflow::XlaExpression
1275 XlaExpression::XlaOp(expression.AsXlaOp(builder), expression.dtype()); in ConvertConstantsToExpressions()
69 auto input_op = arg.AsXlaOp(builder); in Compile()