Searched refs:CustomCallHandler (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/runtime/ |
| D | custom_call.h | 182 class CustomCallHandler; variable
265 std::unique_ptr<CustomCallHandler<checks, Fn, Ts...>> To(Fn fn) {
266 return std::unique_ptr<CustomCallHandler<checks, Fn, Ts...>>(
267 new CustomCallHandler<checks, Fn, Ts...>(
671 class CustomCallHandler : public CustomCall {
788 CustomCallHandler(Fn fn, std::string callee, std::vector<std::string> attrs,
818 constexpr int64_t CustomCallHandler<checks, Fn, Ts...>::kSize;
821 constexpr int64_t CustomCallHandler<checks, Fn, Ts...>::kNumArgs;
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.h | 203 using CustomCallHandler = std::function<StatusOr<Literal>( variable
|