Searched refs:decode_constant_hook_ (Results 1 – 2 of 2) sorted by relevance
134 decode_constant_hook_ = std::move(fn); in RegisterDecodeConstantHook()137 if (decode_constant_hook_) return decode_constant_hook_(input, output); in decode()147 static DecodeConstantHook decode_constant_hook_; variable
208 TensorFlowDialect::DecodeConstantHook TensorFlowDialect::decode_constant_hook_; member in mlir::TF::TensorFlowDialect