Searched defs:UnaryOpsCompositionBase (Results 1 – 1 of 1) sorted by relevance
35 struct UnaryOpsCompositionBase { struct36 using InputBuffer = typename TTypes<T>::ConstFlat;37 using OutputBuffer = typename TTypes<T>::Flat;39 using ComputeFn = void (*)(const InputBuffer&, OutputBuffer*);41 struct ComputeFnRegistration {46 bool HasComputeFn(const string& name) { in HasComputeFn()51 void RegisterComputeFn(const string& name, ComputeFn compute_fn, int cost) { in RegisterComputeFn()59 Status ExportComputeFns(const std::vector<string>& op_names, in ExportComputeFns()75 std::unordered_map<string, ComputeFnRegistration> compute_fns;