Searched refs:xla_op (Results 1 – 6 of 6) sorted by relevance
56 string GetMlirOpString(XlaOp xla_op) { in GetMlirOpString() argument59 xla_builder_.GetValue(xla_op).print(ostream); in GetMlirOpString()
86 for (auto xla_op : ops) { in GetValues() local87 values.push_back(GetValue(xla_op)); in GetValues()
574 for (XlaOp xla_op : operands) { in CreateOp() local575 operand_values.push_back(GetValue(xla_op)); in CreateOp()
231 auto xla_op = in TEST_F() local234 NodeWith(Op("AssignVariableOp"), Inputs(Out(NodeWith()), Out(xla_op))); in TEST_F()
562 ::xla::XlaOp xla_op = xla_op_or.ValueOrDie(); in GetExprForOperand()570 return tensorflow::XlaExpression::XlaOp(xla_op, dtype); in GetExprForOperand()
1218 xla::XlaOp xla_op) -> mlir::LogicalResult { in Lower() argument1220 auto* shape = xla::internal::XlaBuilderFriend::GetInstruction(xla_op) in Lower()1235 xla::internal::XlaBuilderFriend::GetInstruction(xla_op) in Lower()