| 137 | } |
| 138 | |
| 139 | Status XlaOpKernelContext::ConstantInput(int index, |
| 140 | xla::Literal* constant_literal) { |
| 141 | return ConstantInputReshaped( |
| 142 | index, context_->input(index).shape().dim_sizes(), constant_literal); |
| 143 | } |
| 144 | |
| 145 | static xla::StatusOr<int> InputIndex(XlaOpKernelContext* context, |
| 146 | absl::string_view name) { |