| 367 | } |
| 368 | |
| 369 | Status XlaOpKernelContext::InputList(absl::string_view name, |
| 370 | std::vector<xla::XlaOp>* handles, |
| 371 | std::vector<TensorShape>* shapes) { |
| 372 | OpInputList inputs; |
| 373 | TF_RETURN_IF_ERROR(context_->input_list(name, &inputs)); |
| 374 | handles->clear(); |
| 375 | shapes->clear(); |
| 376 | for (const Tensor& input : inputs) { |
| 377 | handles->push_back(CastExpressionFromTensor(input)->AsXlaOp(builder())); |
| 378 | shapes->push_back(input.shape()); |
| 379 | } |
| 380 | return Status::OK(); |
| 381 | } |
| 382 | |
| 383 | Status XlaOpKernelContext::ConstantInputList( |
| 384 | absl::string_view name, std::vector<xla::Literal>* outputs) { |
no test coverage detected